jax-resnet
flaxmodels
jax-resnet | flaxmodels | |
---|---|---|
1 | 1 | |
98 | 222 | |
- | - | |
0.0 | 3.1 | |
almost 2 years ago | 9 months ago | |
Python | Python | |
MIT License | - |
Stars - the number of stars that a project has on GitHub. Growth - month over month growth in stars.
Activity is a relative number indicating how actively a project is being developed. Recent commits have higher weight than older ones.
For example, an activity of 9.0 indicates that a project is amongst the top 10% of the most actively developed projects that we are tracking.
jax-resnet
flaxmodels
-
[P] Training StyleGAN2 in Jax (FFHQ and Anime Faces)
a while ago I posted here regarding a repository containing some pretrained models implemented in Jax/Flax. I decided to add training code for these models. Here is the training code for StyleGAN2: https://github.com/matthias-wright/flaxmodels/tree/main/training/stylegan2
What are some alternatives?
dm-haiku - JAX-based neural network library
stylegan2-pytorch - Implementation of Analyzing and Improving the Image Quality of StyleGAN (StyleGAN 2) in PyTorch
pax - A stateful pytree library for training neural networks.
long-range-arena - Long Range Arena for Benchmarking Efficient Transformers
faceswap - Deepfakes Software For All
jax-models - Unofficial JAX implementations of deep learning research papers
elegy - A High Level API for Deep Learning in JAX
ffhq-dataset - Flickr-Faces-HQ Dataset (FFHQ)
HJxB - Continuous-Time/State/Action Fitted Value Iteration via Hamilton-Jacobi-Bellman (HJB)
transformers - 🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.
DeepFake-Detection - Towards deepfake detection that actually works
prompt-tuning - Original Implementation of Prompt Tuning from Lester, et al, 2021