GitHub repository: tflahaul/jax-vision-transformers
Primary language: Python
Description: Vision transformers with JAX & Flax (ViT, DeiT, LeViT, MAE, ConvPass)