Name		Name	Last commit message	Last commit date
Latest commit History 38 Commits
vit_jax_flax		vit_jax_flax
LICENSE		LICENSE
README.md		README.md
train.py		train.py
vit_jax_flax.ipynb		vit_jax_flax.ipynb

Repository files navigation

vit-jax-flax

Vision Transformer from scratch (JAX/Flax).

Implementation details are described in Vision Transformer from scratch (JAX / Flax)

About

Vision Transformer from scratch (JAX/Flax).

machine-learning deep transformer vision flax jax vision-transformer

Apache-2.0 license

Report repository

Releases

No releases published

Packages

Contributors

Languages