Hugging Face's logo

tensorkelechi
/
dinov2-small-jax

Model card Files
xet
Community

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.

Downloads last month

-

Downloads are not tracked for this model. How to track
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support

Model tree for tensorkelechi/dinov2-small-jax

Base model

facebook/dinov2-small
Finetuned
(20)
this model

Collection including tensorkelechi/dinov2-small-jax