Alternatives and similar repositories for GANs-JAX

Users that are interested in GANs-JAX are comparing it to the libraries listed below

• matthias-wright / flaxmodels
  Pretrained deep learning models for Jax/Flax: StyleGAN2, GPT2, VGG, ResNet, etc.
  ☆265Updated 10 months ago
• yiyixuxu / denoising-diffusion-flax
  Implementing the Denoising Diffusion Probabilistic Model in Flax
  ☆156Updated 3 years ago
• n2cholas / jax-resnet
  Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).
  ☆119Updated 3 years ago
• patil-suraj / stable-diffusion-jax
  ☆91Updated 3 years ago
• moono / stylegan2-tf-2.x
  stylegan2, tensorflow 2, keras subclassing
  ☆102Updated 3 years ago
• kingoflolz / CLIP_JAX
  Contrastive Language-Image Pretraining
  ☆144Updated 3 years ago
• cwkx / GON
  Gradient Origin Networks - a new type of generative model that is able to quickly learn a latent representation without an encoder
  ☆160Updated 5 years ago
• n2cholas / progan-flax
  Flax (JAX) implementation of Progressive Growing of GANs for Improved Quality, Stability, and Variation
  ☆12Updated 4 years ago
• crowsonkb / jax-wavelets
  The 2D discrete wavelet transform for JAX
  ☆44Updated 2 years ago
• crowsonkb / v-diffusion-jax
  v objective diffusion inference code for JAX.
  ☆214Updated 3 years ago
• lucidrains / Adan-pytorch
  Implementation of the Adan (ADAptive Nesterov momentum algorithm) Optimizer in Pytorch
  ☆252Updated 3 years ago
• Rayhane-mamah / Efficient-VDVAE
  Official Pytorch and JAX implementation of "Efficient-VDVAE: Less is more"
  ☆198Updated 3 years ago
• scart97 / Siren-fastai2
  Unofficial implementation of 'Implicit Neural Representations with Periodic Activation Functions'
  ☆52Updated 5 years ago
• unixpickle / vae-textures
  Texture mapping with variational auto-encoders
  ☆40Updated 4 years ago
• HomebrewML / revlib
  Simple and efficient RevNet-Library for PyTorch with XLA and DeepSpeed support and parameter offload
  ☆131Updated 3 years ago
• crowsonkb / esgd
  ESGD-M is a stochastic non-convex second order optimizer, suitable for training deep learning models, for PyTorch.
  ☆56Updated 3 years ago
• patil-suraj / vqgan-jax
  JAX implementation of VQGAN
  ☆91Updated 3 years ago
• gisilvs / AEF
  ☆33Updated 2 years ago
• DarshanDeshpande / jax-models
  Unofficial JAX implementations of deep learning research papers
  ☆161Updated 3 years ago
• google-deepmind / dm_pix
  PIX is an image processing library in JAX, for JAX.
  ☆432Updated 10 months ago
• shyamsn97 / hyper-nn
  Easy Hypernetworks in Pytorch and Jax
  ☆106Updated 3 years ago
• niopeng / HyperRIM
  ☆87Updated 3 years ago
• titu1994 / tf_fourier_features
  Tensorflow 2.0 implementation of Fourier Feature Mapping Networks.
  ☆44Updated 5 years ago
• google-deepmind / jmp
  JMP is a Mixed Precision library for JAX.
  ☆211Updated last year
• AakashKumarNain / diffusion_models
  All about the fundamentals and working of Diffusion Models
  ☆159Updated 3 years ago
• lucidrains / hourglass-transformer-pytorch
  Implementation of Hourglass Transformer, in Pytorch, from Google and OpenAI
  ☆98Updated 4 years ago
• giannisdaras / ilo
  [ICML 2021] Official implementation: Intermediate Layer Optimization for Inverse Problems using Deep Generative Models
  ☆121Updated 3 years ago
• lucidrains / transganformer
  Implementation of TransGanFormer, an all-attention GAN that combines the finding from the recent GanFormer and TransGan paper
  ☆155Updated 4 years ago
• infocusp / diffusion_models
  Minimal standalone example of diffusion model
  ☆163Updated 3 years ago
• EmilienDupont / neural-function-distributions
  Pytorch implementation of Generative Models as Distributions of Functions 🌿
  ☆152Updated 2 years ago