This repository releases pre-trained computer vision models including Vision Transformers and MLP-Mixer architectures. The models were pre-trained on ImageNet and ImageNet-21k datasets. It provides code for fine-tuning these models using JAX and Flax frameworks, enabling transfer learning on custom image classification tasks.