Alternatives and similar repositories for jax-diffusion

Users that are interested in jax-diffusion are comparing it to the libraries listed below

• radarFudan / mamba-minimal-jax
  ☆31Updated 7 months ago
• matthias-wright / jax-fid
  FID computation in Jax/Flax.
  ☆27Updated 11 months ago
• phlippe / jax_trainer
  Lightning-like training API for JAX with Flax
  ☆41Updated 6 months ago
• lucidrains / GAF-microbatch-pytorch
  Implementation of Gradient Agreement Filtering, from Chaubard et al. of Stanford, but for single machine microbatches, in Pytorch
  ☆25Updated 5 months ago
• evanatyourservice / psgd_jax
  Implementation of PSGD optimizer in JAX
  ☆33Updated 5 months ago
• vvvm23 / mamba-jax
  Unofficial but Efficient Implementation of "Mamba: Linear-Time Sequence Modeling with Selective State Spaces" in JAX
  ☆83Updated last year
• kvfrans / jax-flow
  Flow-matching algorithms in JAX
  ☆97Updated 10 months ago
• yilundu / irem_code_release
  ICML 2022: Learning Iterative Reasoning through Energy Minimization
  ☆46Updated 2 years ago
• facebookresearch / SOC-matching
  Code release for "Stochastic Optimal Control Matching"
  ☆35Updated 10 months ago
• GFNOrg / gfn-diffusion
  ☆34Updated 2 months ago
• lucidrains / frame-averaging-pytorch
  Pytorch implementation of a simple way to enable (Stochastic) Frame Averaging for any network
  ☆50Updated 11 months ago
• yilundu / ired_code_release
  ☆65Updated last year
• BirkhoffG / jax-dataloader
  Pytorch-like dataloaders for JAX.
  ☆87Updated last month
• kvfrans / splus
  ☆104Updated 2 weeks ago
• lucidrains / scaling-vin-pytorch
  Exploration into the Scaling Value Iteration Networks paper, from Schmidhuber's group
  ☆36Updated 9 months ago
• google-deepmind / spectral_ssm
  ☆32Updated last year
• lucidrains / blackbox-gradient-sensing
  Implementation and explorations into Blackbox Gradient Sensing (BGS), an evolutionary strategies approach proposed in a Google Deepmind p…
  ☆16Updated 3 weeks ago
• young-geng / mintext
  Minimal but scalable implementation of large language models in JAX
  ☆35Updated 7 months ago
• lucidrains / gateloop-transformer
  Implementation of GateLoop Transformer in Pytorch and Jax
  ☆89Updated last year
• AakashKumarNain / mistral_jax
  This is a port of Mistral-7B model in JAX
  ☆32Updated 11 months ago
• smearle / autoverse
  Generative cellular automaton-like learning environments for RL.
  ☆19Updated 4 months ago
• kvfrans / jax-vqvae-vqgan
  JAX implementation of VQVAE/VQGAN autoencoders (+FSQ)
  ☆30Updated last year
• lucaslingle / mu_transformer
  Transformer with Mu-Parameterization, implemented in Jax/Flax. Supports FSDP on TPU pods.
  ☆30Updated 3 weeks ago
• ml-gde / jflux
  JAX Implementation of Black Forest Labs' Flux.1 family of models
  ☆34Updated 8 months ago
• google-deepmind / md4
  Official Jax Implementation of MD4 Masked Diffusion Models
  ☆106Updated 3 months ago
• brentyi / minGPT-flax
  GPT implementation in Flax
  ☆18Updated 3 years ago
• edwardmilsom / function-space-learning-rates-paper
  Code for the paper "Function-Space Learning Rates"
  ☆20Updated 3 weeks ago
• qsh-zh / pis
  [ICLR 2022] Path integral sampler
  ☆46Updated last year
• dlwh / jax_sourceror
  Turn jitted jax functions back into python source code
  ☆22Updated 6 months ago
• lucidrains / hl-gauss-pytorch
  The Gaussian Histogram Loss (HL-Gauss) proposed by Imani et al. with a few convenient wrappers for regression, in Pytorch
  ☆64Updated 3 weeks ago