soran-ghaderi / torchebmLinks
π Build and train energy-based and diffusion models in PyTorch β‘.
β43Updated this week
Alternatives and similar repositories for torchebm
Users that are interested in torchebm are comparing it to the libraries listed below
Sorting:
- Erwin: A Tree-based Hierarchical Transformer for Large-scale Physical Systems [ICML'25]β101Updated 3 months ago
- code for "Adjoint Sampling: Highly Scalable Diffusion Samplers via Adjoint Matching"β121Updated 2 months ago
- Meta Flow Matching: Integrating Vector Fields on the Wasserstein Manifoldβ64Updated 7 months ago
- β32Updated 2 months ago
- The Superposition of Diffusion Models Using the ItΓ΄ Density Estimatorβ51Updated 6 months ago
- code for "Riemannian Flow Matching on General Geometries".β263Updated last year
- Bare-bones implementations of some generative models in Jax: diffusion, normalizing flows, consistency models, flow matching, (beta)-VAEsβ¦β135Updated last year
- Unofficial implementation of GotenNet, new SOTA 3d equivariant transformer, in Pytorchβ67Updated 6 months ago
- β124Updated last year
- Official repository for "Energy Matching: Unifying Flow Matching and Energy-Based Models for Generative Modeling" [NeurIPS 2025]β163Updated last week
- Clifford-Steerable Convolutional Neural Networks [ICML'24]β49Updated 5 months ago
- Code repository for Trajectory Flow Matchingβ83Updated 11 months ago
- β37Updated 3 months ago
- Flow Annealed Importance Sampling Bootstrap (FAB). ICLR 2023.β65Updated last year
- Official implementation of Fisher-Flow Matching (NeurIPS 2024).β29Updated 11 months ago
- TopoBench is a Python library designed to standardize benchmarking and accelerate research in Topological Deep Learningβ189Updated this week
- Code for the paper https://arxiv.org/abs/2402.04997β96Updated last year
- Official implementation of Metric Flow Matching (NeurIPS 2024)β54Updated last month
- Implementation of Action Matchingβ46Updated 2 years ago
- Code used by the "Clifford Group Equivariant Neural Networks" paper.β84Updated last year
- Improved sampling via learned diffusions (ICLR2024) and an optimal control perspective on diffusion-based generative modeling (TMLR2024)β67Updated 6 months ago
- Code for paper E(n)-Equivariant Topological Neural Networksβ36Updated 10 months ago
- Implementation of Particle Guidance: non-I.I.D. Diverse Sampling with Diffusion Modelsβ71Updated last year
- Score-based generative models for compact manifoldsβ116Updated last year
- Pytorch implementation of a simple way to enable (Stochastic) Frame Averaging for any networkβ51Updated last year
- Reward fine-tuning for Stable Diffusion models based on stochastic optimal control, including Adjoint Matchingβ44Updated 4 months ago
- Lagrangian formulation of Doob's h-transform allowing for efficient rare event samplingβ52Updated 6 months ago
- Geometric Generative Model Tutorial at LoG 2024β27Updated 10 months ago
- Simple Guidance Mechanisms for Discrete Diffusion Modelsβ52Updated 9 months ago
- Flow-matching algorithms in JAXβ105Updated last year