dinov2-small, JAX/Flax port.
I implemented Dinov2 in JAX for TPU inference, and converted the pretrained weights to JAX. Model code here. Only thing here is the pickle file for the converted pytree/state dict or weights.
I implemented Dinov2 in JAX for TPU inference, and converted the pretrained weights to JAX. Model code here. Only thing here is the pickle file for the converted pytree/state dict or weights.