lweitkamp / GANs-JAXLinks
Implementation of several Generative Adversarial Networks in JAX / Flax
☆35Updated 3 years ago
Alternatives and similar repositories for GANs-JAX
Users that are interested in GANs-JAX are comparing it to the libraries listed below
Sorting:
- Pretrained deep learning models for Jax/Flax: StyleGAN2, GPT2, VGG, ResNet, etc.☆264Updated 9 months ago
- Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).☆118Updated 3 years ago
- Implementing the Denoising Diffusion Probabilistic Model in Flax☆156Updated 3 years ago
- Flax (JAX) implementation of Progressive Growing of GANs for Improved Quality, Stability, and Variation☆12Updated 4 years ago
- Tensorflow 2.0 implementation of Fourier Feature Mapping Networks.☆44Updated 5 years ago
- Contrastive Language-Image Pretraining☆144Updated 3 years ago
- ☆33Updated 2 years ago
- ☆91Updated 3 years ago
- A selection of neural network models ported from torchvision for JAX & Flax.☆45Updated 5 months ago
- Gradient Origin Networks - a new type of generative model that is able to quickly learn a latent representation without an encoder☆160Updated 4 years ago
- Official Pytorch and JAX implementation of "Efficient-VDVAE: Less is more"☆198Updated 3 years ago
- The 2D discrete wavelet transform for JAX☆44Updated 2 years ago
- v objective diffusion inference code for JAX.☆214Updated 3 years ago
- Pedagogical codebase for a simplified score-based generative model design, with training loop☆40Updated 4 years ago
- Pytorch implementation of Generative Models as Distributions of Functions 🌿☆152Updated 2 years ago
- ESGD-M is a stochastic non-convex second order optimizer, suitable for training deep learning models, for PyTorch.☆56Updated 3 years ago
- JAX implementation of VQGAN☆91Updated 3 years ago
- Towards Faster and Stabilized GAN Training for High-fidelity Few-shot Image Synthesis☆70Updated 4 years ago
- Unofficial implementation of 'Implicit Neural Representations with Periodic Activation Functions'☆52Updated 5 years ago
- Texture mapping with variational auto-encoders☆40Updated 4 years ago
- Very deep VAEs in JAX/Flax☆46Updated 4 years ago
- Implementation of the Adan (ADAptive Nesterov momentum algorithm) Optimizer in Pytorch☆252Updated 3 years ago
- [CVPR 2021 Oral] Official PyTorch implementation of Soft-IntroVAE from the paper "Soft-IntroVAE: Analyzing and Improving Introspective Va…☆198Updated 3 years ago
- A σ-VAE implementation in PyTorch☆101Updated 4 years ago
- Unofficial JAX implementations of deep learning research papers☆160Updated 3 years ago
- Use Jax functions in Pytorch☆258Updated 2 years ago
- Easy Hypernetworks in Pytorch and Jax☆106Updated 2 years ago
- stylegan2, tensorflow 2, keras subclassing☆102Updated 3 years ago
- Score-Based Generative Modeling with Critically-Damped Langevin Diffusion☆200Updated 2 years ago
- Simple and efficient RevNet-Library for PyTorch with XLA and DeepSpeed support and parameter offload☆131Updated 3 years ago