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.
  ☆254Updated 3 months ago
• yiyixuxu / denoising-diffusion-flax
  Implementing the Denoising Diffusion Probabilistic Model in Flax
  ☆149Updated 2 years ago
• crowsonkb / jax-wavelets
  The 2D discrete wavelet transform for JAX
  ☆43Updated 2 years ago
• n2cholas / jax-resnet
  Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).
  ☆112Updated 3 years ago
• n2cholas / progan-flax
  Flax (JAX) implementation of Progressive Growing of GANs for Improved Quality, Stability, and Variation
  ☆12Updated 4 years ago
• matthias-wright / jax-fid
  FID computation in Jax/Flax.
  ☆27Updated 11 months ago
• rdyro / torch2jax
  Wraps PyTorch code in a JIT-compatible way for JAX. Supports automatically defining gradients for reverse-mode AutoDiff.
  ☆54Updated last month
• conceptofmind / vit-flax
  Implementation of numerous Vision Transformers in Google's JAX and Flax.
  ☆22Updated 2 years ago
• dasayan05 / iclr24_blog_code
  Code repo for ICLR 24 BlogPost titled "Building Diffusion Model's theory from ground up"
  ☆20Updated last year
• gisilvs / AEF
  ☆33Updated 2 years ago
• patil-suraj / stable-diffusion-jax
  ☆89Updated 2 years ago
• titu1994 / tf_fourier_features
  Tensorflow 2.0 implementation of Fourier Feature Mapping Networks.
  ☆42Updated 5 years ago
• ludwigwinkler / JaxLightning
  Running Jax in PyTorch Lightning
  ☆102Updated 6 months ago
• google-deepmind / jmp
  JMP is a Mixed Precision library for JAX.
  ☆203Updated 4 months ago
• scart97 / Siren-fastai2
  Unofficial implementation of 'Implicit Neural Representations with Periodic Activation Functions'
  ☆52Updated 5 years ago
• kingoflolz / CLIP_JAX
  Contrastive Language-Image Pretraining
  ☆143Updated 2 years ago
• rolandgvc / flaxvision
  A selection of neural network models ported from torchvision for JAX & Flax.
  ☆44Updated 4 years ago
• shyamsn97 / hyper-nn
  Easy Hypernetworks in Pytorch and Jax
  ☆101Updated 2 years ago
• TomFrederik / grokking
  Re-implementation of 'Grokking: Generalization beyond overfitting on small algorithmic datasets'
  ☆38Updated 3 years ago
• brandstetter-johannes / LPSDA
  Repo to the paper "Lie Point Symmetry Data Augmentation for Neural PDE Solvers"
  ☆51Updated 2 years ago
• martenlienen / finite-element-networks
  Reference implementation of Finite Element Networks as proposed in "Learning the Dynamics of Physical Systems from Sparse Observations wi…
  ☆69Updated last year
• tum-pbs / PhiML
  Intuitive scientific computing with dimension types for Jax, PyTorch, TensorFlow & NumPy
  ☆90Updated last month
• unixpickle / vae-textures
  Texture mapping with variational auto-encoders
  ☆40Updated 3 years ago
• google-deepmind / dm_pix
  PIX is an image processing library in JAX, for JAX.
  ☆417Updated 3 months ago
• valentingol / GANJax
  Implementation of different GAN in JAX/Haiku.
  ☆14Updated 3 years ago
• kvfrans / jax-flow
  Flow-matching algorithms in JAX
  ☆97Updated 10 months ago
• lucidrains / jax2torch
  Use Jax functions in Pytorch
  ☆242Updated last year
• AakashKumarNain / diffusion_models
  All about the fundamentals and working of Diffusion Models
  ☆158Updated 2 years ago
• teddykoker / learning-to-learn-jax
  JAX implementation of Learning to learn by gradient descent by gradient descent
  ☆27Updated 8 months ago
• samuela / torch2jax
  Run PyTorch in JAX. 🤝
  ☆252Updated 4 months ago