vit-flax (pypi): Vision Transformer, ViT - Flax
Keywords: computer|vision|transformers|flax|jax|in|vision-transformer