Vision Transformer from scratch (JAX/Flax).
Primary LanguageJupyter NotebookApache License 2.0Apache-2.0