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.
  ☆249Updated last year
• yiyixuxu / denoising-diffusion-flax
  Implementing the Denoising Diffusion Probabilistic Model in Flax
  ☆147Updated 2 years ago
• n2cholas / jax-resnet
  Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).
  ☆107Updated 2 years ago
• titu1994 / tf_fourier_features
  Tensorflow 2.0 implementation of Fourier Feature Mapping Networks.
  ☆42Updated 4 years ago
• google-deepmind / jmp
  JMP is a Mixed Precision library for JAX.
  ☆193Updated last month
• n2cholas / progan-flax
  Flax (JAX) implementation of Progressive Growing of GANs for Improved Quality, Stability, and Variation
  ☆12Updated 3 years ago
• crowsonkb / jax-wavelets
  The 2D discrete wavelet transform for JAX
  ☆40Updated 2 years ago
• conceptofmind / vit-flax
  Implementation of numerous Vision Transformers in Google's JAX and Flax.
  ☆22Updated 2 years ago
• lucidrains / jax2torch
  Use Jax functions in Pytorch
  ☆237Updated last year
• unixpickle / vae-textures
  Texture mapping with variational auto-encoders
  ☆40Updated 3 years ago
• scart97 / Siren-fastai2
  Unofficial implementation of 'Implicit Neural Representations with Periodic Activation Functions'
  ☆51Updated 4 years ago
• samuela / torch2jax
  Run PyTorch in JAX. 🤝
  ☆225Updated 3 weeks ago
• matthias-wright / jax-fid
  FID computation in Jax/Flax.
  ☆27Updated 7 months ago
• lucidrains / hourglass-transformer-pytorch
  Implementation of Hourglass Transformer, in Pytorch, from Google and OpenAI
  ☆85Updated 3 years ago
• j-towns / vdvae-jax
  Very deep VAEs in JAX/Flax
  ☆46Updated 3 years ago
• Charl-AI / stochastic-caching
  A tiny library for stochastic dataset caching in PyTorch.
  ☆40Updated last year
• kingoflolz / CLIP_JAX
  Contrastive Language-Image Pretraining
  ☆142Updated 2 years ago
• rdyro / torch2jax
  Wraps PyTorch code in a JIT-compatible way for JAX. Supports automatically defining gradients for reverse-mode AutoDiff.
  ☆46Updated last week
• martenlienen / finite-element-networks
  Reference implementation of Finite Element Networks as proposed in "Learning the Dynamics of Physical Systems from Sparse Observations wi…
  ☆70Updated 9 months ago
• brandstetter-johannes / LPSDA
  Repo to the paper "Lie Point Symmetry Data Augmentation for Neural PDE Solvers"
  ☆49Updated last year
• google / CommonLoopUtils
  CLU lets you write beautiful training loops in JAX.
  ☆333Updated this week
• shyamsn97 / hyper-nn
  Easy Hypernetworks in Pytorch and Jax
  ☆98Updated 2 years ago
• google-deepmind / dm_pix
  PIX is an image processing library in JAX, for JAX.
  ☆407Updated this week
• 4rtemi5 / imax
  Image augmentation library for Jax
  ☆38Updated 11 months ago
• google-deepmind / tf2jax
  ☆112Updated last month
• crowsonkb / v-diffusion-jax
  v objective diffusion inference code for JAX.
  ☆213Updated 2 years ago
• rolandgvc / flaxvision
  A selection of neural network models ported from torchvision for JAX & Flax.
  ☆44Updated 4 years ago
• 8bitmp3 / JAX-Flax-Tutorial-Image-Classification-with-Linen
  How to use the Flax Linen API to build a convolutional neural network model and train it for image classification (using TensorFlow Datas…
  ☆24Updated last year
• teddykoker / learning-to-learn-jax
  JAX implementation of Learning to learn by gradient descent by gradient descent
  ☆27Updated 4 months ago
• valentingol / GANJax
  Implementation of different GAN in JAX/Haiku.
  ☆14Updated 3 years ago