dantp-ai / vit-jax

Implementation of ViT (Vision Transformers) in Jax & Flax

Geek Repo

Github PK Tool