https://github.com/matthias-wright/flaxmodels
Python238
last year
Pretrained deep learning models for Jax/Flax: StyleGAN2, GPT2, VGG, ResNet, etc.
FID computation in Jax/Flax.
Python24apache-2.0
4 months ago
fidflaxfrechet-inception-distance