Alternatives and similar repositories for GANJax

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

• crowsonkb / jax-wavelets
  The 2D discrete wavelet transform for JAX
  ☆43Updated 2 years ago
• lweitkamp / GANs-JAX
  Implementation of several Generative Adversarial Networks in JAX / Flax
  ☆34Updated 3 years ago
• boschresearch / multiplicative-filter-networks
  Source code for Fathony, Sahu, Willmott, & Kolter, "Multiplicative Filter Networks", ICLR 2021.
  ☆94Updated 4 years ago
• flaport / torch_sparse_solve
  A sparse KLU solver for PyTorch.
  ☆65Updated 3 years ago
• White-Link / gpm
  Official implementation of the paper "Unifying GANs and Score-Based Diffusion as Generative Particle Models", NeurIPS 2023
  ☆25Updated 4 months ago
• tumaer / jax-sph
  JAX-SPH: A Differentiable Smoothed Particle Hydrodynamics Framework
  ☆63Updated 4 months ago
• tum-pbs / SIP
  Scale-invariant Learning by Physics Inversion (NeurIPS 2022)
  ☆10Updated 2 years ago
• valentingol / torch_pca
  Principal Component Anlaysis (PCA) in PyTorch.
  ☆21Updated last week
• matthias-wright / jax-fid
  FID computation in Jax/Flax.
  ☆27Updated 10 months ago
• MagnusPetersen / Neural-Cellular-Automata-Image-Manipulation
  Artistic style transfer has been part of the quickly growing AI Art community in recent times. Pioneered by Gatys et al this class of met…
  ☆25Updated 3 years ago
• DiffEqML / pde-dataset-generator
  A tool for generating PDEs ground truth datasets from ARCSim, FEniCS and SU2
  ☆37Updated 3 years ago
• qiauil / torchfsm
  TorchFSM: Fourier Spectral Method with PyTorch
  ☆43Updated this week
• ltatzel / PyTorchHessianFree
  PyTorch implementation of the Hessian-free optimizer
  ☆33Updated 11 months ago
• crowsonkb / mdmm
  The Modified Differential Multiplier Method (MDMM) for PyTorch
  ☆59Updated 4 years ago
• ndahlquist / pytorch-fourier-feature-networks
  Fourier Features Let Networks Learn High Frequency Functions in Low Dimensional Domains
  ☆47Updated 4 years ago
• matthias-wright / flaxmodels
  Pretrained deep learning models for Jax/Flax: StyleGAN2, GPT2, VGG, ResNet, etc.
  ☆252Updated last month
• belkakari / implicit-kan
  Kolmogorov-Arnold networks (KAN) as implicit functions (like NeRF but simpler)
  ☆14Updated last year
• greydanus / optimize_wing
  We simulate a wind tunnel, place a rectangular occlusion in it, and then use gradient descent to turn the occlusion into a wing.
  ☆26Updated 4 years ago
• kklemon / gon-pytorch
  ☆16Updated 3 years ago
• tum-pbs / DMCF
  Guaranteed Conservation of Momentum for Learning Particle-based Fluid Dynamics (NeurIPS '22)
  ☆54Updated last year
• samuelepapa / fit-a-nef
  🚀 A powerful library for efficient training of Neural Fields at scale.
  ☆29Updated last year
• fabiodimarco / torch-levenberg-marquardt
  PyTorch implementation of Levenberg-Marquardt training algorithm
  ☆60Updated last month
• yzshi5 / OpFlow
  codes for "Universal Functional Regression with Neural Operator Flows"
  ☆13Updated 7 months ago
• rdyro / torch2jax
  Wraps PyTorch code in a JIT-compatible way for JAX. Supports automatically defining gradients for reverse-mode AutoDiff.
  ☆52Updated 2 weeks ago
• pnkraemer / probdiffeq
  Probabilistic solvers for differential equations in JAX. Adaptive ODE solvers with calibration, state-space model factorisations, and cus…
  ☆46Updated last month
• gisilvs / AEF
  ☆33Updated 2 years ago
• tum-pbs / apebench
  [Neurips 2024] A benchmark suite for autoregressive neural emulation of PDEs. (≥46 PDEs in 1D, 2D, 3D; Differentiable Physics; Unrolled T…
  ☆70Updated 6 months ago
• n2cholas / progan-flax
  Flax (JAX) implementation of Progressive Growing of GANs for Improved Quality, Stability, and Variation
  ☆12Updated 3 years ago
• nshepperd / jax-guided-diffusion
  jax version of clip guided diffusion scripts
  ☆90Updated last year
• unixpickle / vae-textures
  Texture mapping with variational auto-encoders
  ☆40Updated 3 years ago