flaxmodels (pypi): A collection of pretrained models in Flax.
Keywords: deep-learning|flax|gpt2|jax|pretrained-models|resnet|stylegan2|vgg