GitHub repository: rolandgvc/flaxvision
Primary language: Python
Description: A selection of neural network models ported from torchvision for JAX & Flax.