flaxmodels

Pretrained deep learning models for Jax/Flax: StyleGAN2, GPT2, VGG, ResNet, etc.

License

Pretrained deep learning models for Jax/Flax: StyleGAN2, GPT2, VGG, ResNet, etc.

Creator

matthias-wright

Related apps

jax-fid

FID computation in Jax/Flax.

Python21apache-2.0

9 months ago

fidflaxfrechet-inception-distance