satojkovic / vit-jax-flax Public

Notifications You must be signed in to change notification settings
Fork 2
Star 7

Vision Transformer from scratch (JAX/Flax).

Apache-2.0 license

7 stars 2 forks Branches Tags Activity

Notifications

Name		Name	Last commit message	Last commit date
Latest commit History 36 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

No packages published

Languages