https://github.com/matthias-wright/flaxmodels

Pretrained deep learning models for Jax/Flax: StyleGAN2, GPT2, VGG, ResNet, etc.
https://github.com/matthias-wright/flaxmodels

deep-learning flax gpt2 jax pretrained-models resnet stylegan2 vgg

Last synced: 8 days ago
JSON representation

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

• Host: GitHub
• URL: https://github.com/matthias-wright/flaxmodels
• Owner: matthias-wright
• Created: 2021-05-16T13:34:20.000Z (almost 4 years ago)
• Default Branch: main
• Last Pushed: 2025-03-21T10:37:09.000Z (29 days ago)
• Last Synced: 2025-04-04T12:56:56.904Z (15 days ago)
• Topics: deep-learning, flax, gpt2, jax, pretrained-models, resnet, stylegan2, vgg
• Language: Python
• Homepage: https://github.com/matthias-wright/flaxmodels
• Size: 118 MB
• Stars: 249
• Watchers: 5
• Forks: 25
• Open Issues: 3
• Metadata Files:
  • Readme: README.md

Awesome Lists containing this project

• awesome-jax - flaxmodels - Pretrained models for Jax/Flax. <img src="https://img.shields.io/github/stars/matthias-wright/flaxmodels?style=social" align="center"> (Libraries / New Libraries)