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.
  ☆260Updated 6 months ago
• yiyixuxu / denoising-diffusion-flax
  Implementing the Denoising Diffusion Probabilistic Model in Flax
  ☆150Updated 2 years ago
• n2cholas / jax-resnet
  Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).
  ☆114Updated 3 years ago
• kingoflolz / CLIP_JAX
  Contrastive Language-Image Pretraining
  ☆143Updated 3 years ago
• crowsonkb / jax-wavelets
  The 2D discrete wavelet transform for JAX
  ☆42Updated 2 years ago
• n2cholas / progan-flax
  Flax (JAX) implementation of Progressive Growing of GANs for Improved Quality, Stability, and Variation
  ☆12Updated 4 years ago
• unixpickle / vae-textures
  Texture mapping with variational auto-encoders
  ☆40Updated 4 years ago
• google-deepmind / dm_pix
  PIX is an image processing library in JAX, for JAX.
  ☆422Updated 7 months ago
• DarshanDeshpande / jax-models
  Unofficial JAX implementations of deep learning research papers
  ☆156Updated 3 years ago
• patil-suraj / stable-diffusion-jax
  ☆89Updated 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
  ☆159Updated 4 years ago
• patil-suraj / vqgan-jax
  JAX implementation of VQGAN
  ☆91Updated 3 years ago
• lucidrains / Adan-pytorch
  Implementation of the Adan (ADAptive Nesterov momentum algorithm) Optimizer in Pytorch
  ☆252Updated 3 years ago
• AakashKumarNain / diffusion_models
  All about the fundamentals and working of Diffusion Models
  ☆159Updated 2 years ago
• google-deepmind / jmp
  JMP is a Mixed Precision library for JAX.
  ☆207Updated 8 months ago
• crowsonkb / v-diffusion-jax
  v objective diffusion inference code for JAX.
  ☆214Updated 3 years ago
• lucidrains / jax2torch
  Use Jax functions in Pytorch
  ☆251Updated 2 years ago
• gisilvs / AEF
  ☆33Updated 2 years ago
• shyamsn97 / hyper-nn
  Easy Hypernetworks in Pytorch and Jax
  ☆105Updated 2 years ago
• Rayhane-mamah / Efficient-VDVAE
  Official Pytorch and JAX implementation of "Efficient-VDVAE: Less is more"
  ☆196Updated 3 years ago
• scart97 / Siren-fastai2
  Unofficial implementation of 'Implicit Neural Representations with Periodic Activation Functions'
  ☆52Updated 5 years ago
• samuela / torch2jax
  Run PyTorch in JAX. 🤝
  ☆298Updated last month
• lucidrains / siren-pytorch
  Pytorch implementation of SIREN - Implicit Neural Representations with Periodic Activation Function
  ☆504Updated 2 years ago
• EmilienDupont / neural-function-distributions
  Pytorch implementation of Generative Models as Distributions of Functions 🌿
  ☆151Updated last year
• rasmusbergpalm / vnca
  ☆54Updated 2 years ago
• beresandras / clear-diffusion-keras
  Implementation of denoising diffusion models with schedules, improved sampling, and other extensions using Keras.
  ☆120Updated last year
• HomebrewML / revlib
  Simple and efficient RevNet-Library for PyTorch with XLA and DeepSpeed support and parameter offload
  ☆131Updated 3 years ago
• nv-tlabs / CLD-SGM
  Score-Based Generative Modeling with Critically-Damped Langevin Diffusion
  ☆196Updated last year
• crowsonkb / esgd
  ESGD-M is a stochastic non-convex second order optimizer, suitable for training deep learning models, for PyTorch.
  ☆56Updated 3 years ago
• titu1994 / tf_fourier_features
  Tensorflow 2.0 implementation of Fourier Feature Mapping Networks.
  ☆43Updated 5 years ago