Alternatives and similar repositories for GANJax:

Users that are interested in GANJax are comparing it to the libraries listed below

• lweitkamp / GANs-JAX
  Implementation of several Generative Adversarial Networks in JAX / Flax
  ☆33Updated 2 years ago
• crowsonkb / jax-wavelets
  The 2D discrete wavelet transform for JAX
  ☆40Updated 2 years ago
• rakhimovv / voronoi-interp-torch
  ☆13Updated 3 years ago
• flaport / torch_sparse_solve
  A sparse KLU solver for PyTorch.
  ☆62Updated 3 years ago
• tum-pbs / ConFIG
  [ICLR2025 Spotlight] Official implementation of Conflict-Free Inverse Gradients Method
  ☆41Updated last week
• ltatzel / PyTorchHessianFree
  PyTorch implementation of the Hessian-free optimizer
  ☆34Updated 8 months ago
• fabiodimarco / torch-levenberg-marquardt
  PyTorch implementation of Levenberg-Marquardt training algorithm
  ☆48Updated 3 weeks ago
• matthias-wright / jax-fid
  FID computation in Jax/Flax.
  ☆27Updated 7 months ago
• crowsonkb / mdmm
  The Modified Differential Multiplier Method (MDMM) for PyTorch
  ☆57Updated 3 years ago
• probabilists / azula
  Diffusion models in PyTorch
  ☆93Updated this week
• tumaer / jax-sph
  JAX-SPH: A Differentiable Smoothed Particle Hydrodynamics Framework
  ☆59Updated last month
• pharmapsychotic / iadb-fun
  ☆14Updated last year
• crowsonkb / mdmm-jax
  Gradient-based constrained optimization for JAX
  ☆29Updated 2 years ago
• maxxxzdn / clifford-group-equivariant-cnns
  Code repository of the paper "Clifford-Steerable Convolutional Neural Networks"
  ☆47Updated 6 months ago
• cooper-org / cooper
  A general-purpose, deep learning-first library for constrained optimization in PyTorch
  ☆112Updated 2 weeks ago
• francois-rozet / papers-101
  Implementation of papers in 101 lines of code.
  ☆18Updated last year
• kvfrans / jax-flow
  Flow-matching algorithms in JAX
  ☆85Updated 6 months ago
• DiffEqML / pde-dataset-generator
  A tool for generating PDEs ground truth datasets from ARCSim, FEniCS and SU2
  ☆35Updated 3 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…
  ☆70Updated 9 months ago
• samuela / torch2jax
  Run PyTorch in JAX. 🤝
  ☆225Updated 3 weeks ago
• rdyro / torch2jax
  Wraps PyTorch code in a JIT-compatible way for JAX. Supports automatically defining gradients for reverse-mode AutoDiff.
  ☆46Updated last week
• patrick-kidger / sympy2jax
  Turn SymPy expressions into trainable JAX expressions.
  ☆330Updated last month
• tum-pbs / half-inverse-gradients
  Source code for the ICLR'22 paper on "Half-Inverse Gradients"
  ☆18Updated 2 years ago
• yklcs / jaxsplat
  3D Gaussian Splatting in JAX
  ☆57Updated 9 months ago
• facebookincubator / flowtorch
  This library would form a permanent home for reusable components for deep probabilistic programming. The library would form and harness a…
  ☆306Updated 3 months ago
• diffprog / code
  Code for the book "The Elements of Differentiable Programming".
  ☆74Updated 3 weeks ago
• valentingol / torch_pca
  Principal Component Anlaysis (PCA) in PyTorch.
  ☆19Updated 4 months ago
• ludwigwinkler / JaxLightning
  Running Jax in PyTorch Lightning
  ☆89Updated 2 months ago
• rubenohana / Gibbs-Diffusion
  Code of the paper "Listening to the Noise: Blind Denoising with Gibbs Diffusion"
  ☆33Updated 8 months ago