jvt (pypi): Vision transformers with JAX & Flax
Keywords: artificial intelligence|computer vision|jax|transformers