---
license: apache-2.0
base_model:
- facebook/dinov2-small
---

# dinov2-small, JAX/Flax port.
I implemented Dinov2 in JAX for TPU inference, and converted the pretrained weights to JAX.
[Model code here](https://github.com/kelechi-c/snowpark/blob/main/dino_v2_jax/dinov2.py). 
Only thing here is the pickle file for the converted pytree/state dict or weights.