Alternatives and similar repositories for jax-diffusion:

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

• phlippe / jax_trainer
  Lightning-like training API for JAX with Flax
  ☆36Updated last month
• kvfrans / jax-flow
  Flow-matching algorithms in JAX
  ☆82Updated 5 months ago
• vvvm23 / mamba-jax
  Unofficial but Efficient Implementation of "Mamba: Linear-Time Sequence Modeling with Selective State Spaces" in JAX
  ☆82Updated 11 months ago
• lucidrains / frame-averaging-pytorch
  Pytorch implementation of a simple way to enable (Stochastic) Frame Averaging for any network
  ☆48Updated 5 months ago
• facebookresearch / SOC-matching
  Code release for "Stochastic Optimal Control Matching"
  ☆30Updated 5 months ago
• radarFudan / mamba-minimal-jax
  ☆31Updated last month
• liusulin / DDPD
  Code for Paper "Think While You Generate: Discrete Diffusion with Planned Denoising"
  ☆35Updated 2 months ago
• google-deepmind / md4
  Code for https://arxiv.org/abs/2406.04329
  ☆51Updated last month
• BirkhoffG / jax-dataloader
  Pytorch-like dataloaders in JAX.
  ☆67Updated 3 months ago
• juliusberner / sde_sampler
  Improved sampling via learned diffusions (ICLR2024) and an optimal control perspective on diffusion-based generative modeling (TMLR2024)
  ☆56Updated 4 months ago
• lucidrains / gotennet-pytorch
  Implementation of GotenNet, new SOTA 3d equivariant transformer, in Pytorch
  ☆53Updated last week
• zacharyhorvitz / Fk-Diffusion-Steering
  A general framework for inference-time scaling and steering of diffusion models with arbitrary rewards.
  ☆40Updated this week
• helblazer811 / diffusion-visualizations
  Visualizations of the theory behind diffusion models.
  ☆77Updated 9 months ago
• lollcat / fab-torch
  Flow Annealed Importance Sampling Bootstrap (FAB). ICLR 2023.
  ☆53Updated 10 months ago
• louaaron / Reflected-Diffusion
  [ICML 2023] Reflected Diffusion Models (https://arxiv.org/abs/2304.04740)
  ☆157Updated last year
• ling-pan / GAFN
  ☆23Updated 9 months ago
• lucidrains / scaling-vin-pytorch
  Exploration into the Scaling Value Iteration Networks paper, from Schmidhuber's group
  ☆36Updated 3 months ago
• lucidrains / dreamerv3-pytorch
  Implementation of DreamerV3 in Pytorch
  ☆42Updated 2 months ago
• lucidrains / gateloop-transformer
  Implementation of GateLoop Transformer in Pytorch and Jax
  ☆87Updated 7 months ago
• nZhangx / TrajectoryFlowMatching
  Code repository for Trajectory Flow Matching
  ☆50Updated 2 months ago
• facebookresearch / lagrangian-ot
  Neural Optimal Transport with Lagrangian Costs
  ☆50Updated 6 months ago
• yilundu / irem_code_release
  ICML 2022: Learning Iterative Reasoning through Energy Minimization
  ☆44Updated last year
• davisyoshida / lorax
  LoRA for arbitrary JAX models and functions
  ☆135Updated 10 months ago
• yilundu / ired_code_release
  ☆49Updated 7 months ago
• lucidrains / jax2torch
  Use Jax functions in Pytorch
  ☆231Updated last year
• lucidrains / autoregressive-linear-attention-cuda
  CUDA implementation of autoregressive linear attention, with all the latest research findings
  ☆44Updated last year
• necludov / wl-mechanics
  ☆31Updated 7 months ago
• zdhNarsil / Diffusion-Generative-Flow-Samplers
  PyTorch implementation for our ICLR 2024 paper "Diffusion Generative Flow Samplers: Improving learning signals through partial trajectory…
  ☆22Updated last year
• yilundu / reduce_reuse_recycle
  ICML 2023: Reduce, Reuse, Recycle: Composing Energy-Based Diffusion Models with MCMC
  ☆136Updated 3 months ago
• dome272 / Flow-Matching
  My take on Flow Matching
  ☆31Updated last week