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.
  ☆245Updated last year
• n2cholas / jax-resnet
  Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).
  ☆106Updated 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.
  ☆189Updated this week
• yiyixuxu / denoising-diffusion-flax
  Implementing the Denoising Diffusion Probabilistic Model in Flax
  ☆144Updated 2 years ago
• scart97 / Siren-fastai2
  Unofficial implementation of 'Implicit Neural Representations with Periodic Activation Functions'
  ☆51Updated 4 years ago
• gisilvs / AEF
  ☆33Updated last year
• crowsonkb / jax-wavelets
  The 2D discrete wavelet transform for JAX
  ☆40Updated last year
• j-towns / vdvae-jax
  Very deep VAEs in JAX/Flax
  ☆46Updated 3 years ago
• lucidrains / jax2torch
  Use Jax functions in Pytorch
  ☆232Updated last year
• rdyro / torch2jax
  Wraps PyTorch code in a JIT-compatible way for JAX. Supports automatically defining gradients for reverse-mode AutoDiff.
  ☆44Updated last week
• rolandgvc / flaxvision
  A selection of neural network models ported from torchvision for JAX & Flax.
  ☆44Updated 4 years ago
• n2cholas / progan-flax
  Flax (JAX) implementation of Progressive Growing of GANs for Improved Quality, Stability, and Variation
  ☆12Updated 3 years ago
• cgarciae / ciclo
  A functional training loops library for JAX
  ☆86Updated 11 months ago
• unixpickle / vae-textures
  Texture mapping with variational auto-encoders
  ☆40Updated 3 years ago
• ucl-bug / jaxdf
  A JAX-based research framework for writing differentiable numerical simulators with arbitrary discretizations
  ☆124Updated 4 months ago
• samuela / torch2jax
  Run PyTorch in JAX. 🤝
  ☆217Updated 3 weeks ago
• kvfrans / jax-flow
  Flow-matching algorithms in JAX
  ☆83Updated 5 months ago
• matthias-wright / jax-fid
  FID computation in Jax/Flax.
  ☆26Updated 6 months ago
• google / CommonLoopUtils
  CLU lets you write beautiful training loops in JAX.
  ☆329Updated this week
• yklcs / jaxsplat
  3D Gaussian Splatting in JAX
  ☆55Updated 8 months ago
• google-deepmind / tf2jax
  ☆111Updated 3 weeks ago
• stanfordmlgroup / disentanglement
  Official repository for our ICLR 2021 paper Evaluating the Disentanglement of Deep Generative Models with Manifold Topology
  ☆35Updated 3 years ago
• CW-Huang / CP-Flow
  Convex potential flows
  ☆82Updated 3 years ago
• shreyansh26 / ML-Optimizers-JAX
  Toy implementations of some popular ML optimizers using Python/JAX
  ☆43Updated 3 years ago
• crowsonkb / mdmm
  The Modified Differential Multiplier Method (MDMM) for PyTorch
  ☆55Updated 3 years ago
• google / tree-math
  Mathematical operations for JAX pytrees
  ☆198Updated last month
• paganpasta / eqxvision
  A Python package of computer vision models for the Equinox ecosystem.
  ☆103Updated 6 months ago
• ludwigwinkler / JaxLightning
  Running Jax in PyTorch Lightning
  ☆86Updated last month
• lucidrains / all-normalization-transformer
  A simple Transformer where the softmax has been replaced with normalization
  ☆19Updated 4 years ago