Pretrained models for Jax/Flax: StyleGAN2, GPT2, VGG, ResNet, etc. - View it on GitHub