flaxmodels
Pretrained deep learning models for Jax/Flax: StyleGAN2, GPT2, VGG, ResNet, etc.
236 stars
6 watching
25 forks
Language: Python
last commit: about 1 year ago
Linked from 1 awesome list
deep-learningflaxgpt2jaxpretrained-modelsresnetstylegan2vgg
n2cholas/awesome-jax