tensorkelechi
/

dinov2-small-jax

Model card Files Files and versions 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

(9)

this model

Collection including tensorkelechi/dinov2-small-jax

snowpark

JAX ports of popular models(audio, vision, llm, diffusion, etc) • 1 item • Updated Feb 28