Alternatives and similar repositories for torchebm

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

• facebookresearch / adjoint_sampling
  code for "Adjoint Sampling: Highly Scalable Diffusion Samplers via Adjoint Matching"
  ☆101Updated last month
• maxxxzdn / erwin
  Erwin: A Tree-based Hierarchical Transformer for Large-scale Physical Systems [ICML'25]
  ☆70Updated last week
• kvfrans / jax-flow
  Flow-matching algorithms in JAX
  ☆97Updated 10 months ago
• DoronHav / WassersteinFlowMatching
  ☆28Updated last week
• necludov / super-diffusion
  The Superposition of Diffusion Models Using the Itô Density Estimator
  ☆45Updated 3 months ago
• maxxxzdn / clifford-group-equivariant-cnns
  Clifford-Steerable Convolutional Neural Networks [ICML'24]
  ☆47Updated last month
• david-knigge / enf-min-jax
  A minimal implementation of Equivariant Neural Fields (https://arxiv.org/abs/2406.05753).
  ☆25Updated 4 months ago
• lindermanlab / elk
  Scalable and Stable Parallelization of Nonlinear RNNS
  ☆17Updated 5 months ago
• senya-ashukha / simple-equivariant-gnn
  A short and easy PyTorch implementation of E(n) Equivariant Graph Neural Networks
  ☆131Updated 3 years ago
• emilemathieu / escnn_jax
  Equivariant Steerable CNNs Library for Pytorch https://quva-lab.github.io/escnn/
  ☆30Updated 2 years ago
• DavidRuhe / clifford-group-equivariant-neural-networks
  Code used by the "Clifford Group Equivariant Neural Networks" paper.
  ☆83Updated last year
• lazaratan / meta-flow-matching
  Meta Flow Matching: Integrating Vector Fields on the Wasserstein Manifold
  ☆55Updated 3 months ago
• necludov / jam
  Implementation of Action Matching
  ☆44Updated 2 years ago
• ebekkers / ponita
  Fast, Expressive SE(n) Equivariant Networks through Weight-Sharing in Position-Orientation Space.
  ☆84Updated last year
• fal-ai-community / minDDPD
  ☆33Updated 5 months ago
• HannesStark / dirichlet-flow-matching
  ☆115Updated last year
• maxxxzdn / implicit-steerable-kernels
  Implicit Convolutional Kernels for Steerable CNNs [NeurIPS'23]
  ☆28Updated 4 months ago
• smsharma / minified-generative-models
  Bare-bones implementations of some generative models in Jax: diffusion, normalizing flows, consistency models, flow matching, (beta)-VAEs…
  ☆130Updated last year
• akandykeller / NeuralWaveMachines
  Official Implementation of the ICML 2023 paper: "Neural Wave Machines: Learning Spatiotemporally Structured Representations with Locally …
  ☆72Updated 2 years ago
• plainerman / Variational-Doob
  Lagrangian formulation of Doob's h-transform allowing for efficient rare event sampling
  ☆49Updated 3 months ago
• juliusberner / sde_sampler
  Improved sampling via learned diffusions (ICLR2024) and an optimal control perspective on diffusion-based generative modeling (TMLR2024)
  ☆62Updated 3 months ago
• Iolo-Jones / ManifoldDiffusionGeometry
  Code for "Manifold Diffusion Geometry: Curvature, Tangent Spaces, and Dimension"
  ☆44Updated 7 months ago
• microsoft / two-for-one-diffusion
  This Denoising Force Field (DFF) codebase provides a Pytorch framework for the method presented in Two for one: Diffusion models and forc…
  ☆62Updated last year
• itsdaniele / jeometric
  Graph neural networks in JAX.
  ☆67Updated last year
• facebookresearch / lagrangian-ot
  Neural Optimal Transport with Lagrangian Costs
  ☆56Updated last month
• katalinic / sdflows
  JAX exponential map normalising flows on sphere
  ☆17Updated 4 years ago
• lucidrains / gotennet-pytorch
  Unofficial implementation of GotenNet, new SOTA 3d equivariant transformer, in Pytorch
  ☆63Updated 2 months ago
• facebookresearch / riemannian-fm
  code for "Riemannian Flow Matching on General Geometries".
  ☆234Updated last year
• tychovdo / noethers-razor
  Code for NeurIPS 2024 paper: "Noether's razor: Learning Conserved Quantities" by Tycho F. A. van der Ouderaa, Mark van der Wilk, Pim de H…
  ☆10Updated 8 months ago
• lollcat / fab-torch
  Flow Annealed Importance Sampling Bootstrap (FAB). ICLR 2023.
  ☆58Updated last year