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 is not currently available via any of the supported Inference Providers.
The model cannot be deployed to the HF Inference API: The model has no library tag.

Model tree for tensorkelechi/dinov2-small-jax

Finetuned
(9)
this model

Collection including tensorkelechi/dinov2-small-jax