conceptofmind / vit-flax

Implementation of numerous Vision Transformers in Google's JAX and Flax.
20Updated 2 years ago

Related projects

Alternatives and complementary repositories for vit-flax