Vision Transformer from scratch (JAX/Flax). Contribute to satojkovic/vit-jax-flax development by creating an account on GitHub.