File size: 361 Bytes
4ac6151 |
1 2 3 4 5 6 7 8 9 10 |
---
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. |