Alternatives and similar repositories for torchebm

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

• maxxxzdn / erwin
  Erwin: A Tree-based Hierarchical Transformer for Large-scale Physical Systems [ICML'25]
  ☆110Updated 3 months ago
• lazaratan / meta-flow-matching
  Meta Flow Matching: Integrating Vector Fields on the Wasserstein Manifold
  ☆69Updated 10 months ago
• necludov / super-diffusion
  The Superposition of Diffusion Models Using the Itô Density Estimator
  ☆51Updated 10 months ago
• facebookresearch / adjoint_sampling
  code for "Adjoint Sampling: Highly Scalable Diffusion Samplers via Adjoint Matching"
  ☆129Updated 6 months ago
• lucidrains / gotennet-pytorch
  Unofficial implementation of GotenNet, new SOTA 3d equivariant transformer, in Pytorch
  ☆68Updated 9 months ago
• DoronHav / WassersteinFlowMatching
  ☆37Updated this week
• HannesStark / dirichlet-flow-matching
  ☆130Updated last year
• m1balcerak / EnergyMatching
  [NeurIPS 2025] Official repository for "Energy Matching: Unifying Flow Matching and Energy-Based Models for Generative Modeling"
  ☆198Updated 3 months ago
• facebookresearch / riemannian-fm
  code for "Riemannian Flow Matching on General Geometries".
  ☆283Updated last year
• smsharma / minified-generative-models
  Bare-bones implementations of some generative models in Jax: diffusion, normalizing flows, consistency models, flow matching, (beta)-VAEs…
  ☆141Updated 2 years ago
• DavidRuhe / clifford-group-equivariant-neural-networks
  Code used by the "Clifford Group Equivariant Neural Networks" paper.
  ☆90Updated last year
• martaskrt / fkc-diffusion
  ☆44Updated 7 months ago
• maxxxzdn / clifford-group-equivariant-cnns
  Clifford-Steerable Convolutional Neural Networks [ICML'24]
  ☆52Updated 8 months ago
• lollcat / fab-torch
  Flow Annealed Importance Sampling Bootstrap (FAB). ICLR 2023.
  ☆66Updated last year
• lucidrains / frame-averaging-pytorch
  Pytorch implementation of a simple way to enable (Stochastic) Frame Averaging for any network
  ☆51Updated last year
• nZhangx / TrajectoryFlowMatching
  Code repository for Trajectory Flow Matching
  ☆99Updated last year
• kantamasuki / RGDM
  Implementations of the renormalization group-based diffusion model (RGDM).
  ☆15Updated 10 months ago
• gcorso / particle-guidance
  Implementation of Particle Guidance: non-I.I.D. Diverse Sampling with Diffusion Models
  ☆77Updated 2 years ago
• kkapusniak / metric-flow-matching
  Official implementation of Metric Flow Matching (NeurIPS 2024)
  ☆55Updated 5 months ago
• ebekkers / ponita
  Fast, Expressive SE(n) Equivariant Networks through Weight-Sharing in Position-Orientation Space.
  ☆88Updated last year
• plainerman / Variational-Doob
  Lagrangian formulation of Doob's h-transform allowing for efficient rare event sampling
  ☆54Updated 10 months ago
• olsdavis / fisher-flow
  Official implementation of Fisher-Flow Matching (NeurIPS 2024).
  ☆36Updated last year
• andrew-cr / discrete_flow_models
  Code for the paper https://arxiv.org/abs/2402.04997
  ☆103Updated last year
• jarridrb / DEM
  Code for the paper Iterated Denoising Energy Matching for Sampling from Boltzmann Densities.
  ☆68Updated 9 months ago
• juliusberner / sde_sampler
  Improved sampling via learned diffusions (ICLR2024) and an optimal control perspective on diffusion-based generative modeling (TMLR2024)
  ☆71Updated 10 months ago
• facebookresearch / lagrangian-ot
  Neural Optimal Transport with Lagrangian Costs
  ☆61Updated 8 months ago
• maxxxzdn / implicit-steerable-kernels
  Implicit Convolutional Kernels for Steerable CNNs [NeurIPS'23]
  ☆30Updated 11 months ago
• nmboffi / flow-maps
  Official codebase for the paper "How to build a consistency model: Learning flow maps via self-distillation" (NeurIPS 2025).
  ☆70Updated 3 months ago
• masa-ue / AlignInversePro
  Inference-Time Alignment in Protein Diffusion Models
  ☆51Updated 3 weeks ago
• microsoft / soc-fine-tuning-sd
  Reward fine-tuning for Stable Diffusion models based on stochastic optimal control, including Adjoint Matching
  ☆62Updated 8 months ago