soran-ghaderi / torchebmLinks
π Build and train energy-based and diffusion models in PyTorch β‘.
β57Updated 2 weeks ago
Alternatives and similar repositories for torchebm
Users that are interested in torchebm are comparing it to the libraries listed below
Sorting:
- code for "Adjoint Sampling: Highly Scalable Diffusion Samplers via Adjoint Matching"β125Updated 4 months ago
- Erwin: A Tree-based Hierarchical Transformer for Large-scale Physical Systems [ICML'25]β109Updated last month
- Meta Flow Matching: Integrating Vector Fields on the Wasserstein Manifoldβ69Updated 8 months ago
- β126Updated last year
- Unofficial implementation of GotenNet, new SOTA 3d equivariant transformer, in Pytorchβ67Updated 7 months ago
- The Superposition of Diffusion Models Using the ItΓ΄ Density Estimatorβ51Updated 8 months ago
- Bare-bones implementations of some generative models in Jax: diffusion, normalizing flows, consistency models, flow matching, (beta)-VAEsβ¦β137Updated last year
- Pytorch implementation of a simple way to enable (Stochastic) Frame Averaging for any networkβ51Updated last year
- β40Updated 5 months ago
- Lagrangian formulation of Doob's h-transform allowing for efficient rare event samplingβ53Updated 8 months ago
- Implementation of Particle Guidance: non-I.I.D. Diverse Sampling with Diffusion Modelsβ72Updated 2 years ago
- code for "Riemannian Flow Matching on General Geometries".β273Updated last year
- β34Updated 3 months ago
- Code for the paper LaM-SLidE - Latent Space Modeling of Spatial Dynamical Systems via Linked Entitiesβ25Updated 6 months ago
- Code repository for Trajectory Flow Matchingβ90Updated last year
- Official implementation of Fisher-Flow Matching (NeurIPS 2024).β33Updated last year
- Code used by the "Clifford Group Equivariant Neural Networks" paper.β88Updated last year
- [NeurIPS 2025] Official repository for "Energy Matching: Unifying Flow Matching and Energy-Based Models for Generative Modeling"β177Updated last month
- Code for the paper https://arxiv.org/abs/2402.04997β99Updated last year
- This Denoising Force Field (DFF) codebase provides a Pytorch framework for the method presented in Two for one: Diffusion models and forcβ¦β66Updated last year
- Flow Annealed Importance Sampling Bootstrap (FAB). ICLR 2023.β66Updated last year
- β37Updated last year
- Discrete Bayesian optimization with LLMs, PEFT finetuning methods, and the Laplace approximation.β21Updated last year
- β16Updated 6 months ago
- Code for the paper Iterated Denoising Energy Matching for Sampling from Boltzmann Densities.β66Updated 7 months ago
- Geometric Generative Model Tutorial at LoG 2024β27Updated last year
- Improved sampling via learned diffusions (ICLR2024) and an optimal control perspective on diffusion-based generative modeling (TMLR2024)β70Updated 8 months ago
- Uncover meaningful structures of latent spaces learned by generative models with flows!β42Updated last year
- A short and easy PyTorch implementation of E(n) Equivariant Graph Neural Networksβ139Updated 3 years ago
- Implementation of the Equiformer, SE3/E3 equivariant attention network that reaches new SOTA, and adopted for use by EquiFold for proteinβ¦β278Updated 11 months ago