Vision Transformers in JAX/Flax(ViT-Flax)
main
API reference
vit_flax package
Vision Transformers in JAX/Flax(ViT-Flax)
Docs
»
Overview: module code
All modules for which code is available
flax.nn.base
vit_flax.layers
vit_flax.vit