dinov2-small-jax / README.md

tensorkelechi

Create README.md

4ac6151 verified about 2 months ago

preview code

raw

history blame contribute delete

361 Bytes

metadata

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. Only thing here is the pickle file for the converted pytree/state dict or weights.