soran-ghaderi / torchebmLinks
π Build and train energy-based and diffusion models in PyTorch β‘.
β55Updated last week
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"β122Updated 3 months ago
- The Superposition of Diffusion Models Using the ItΓ΄ Density Estimatorβ51Updated 7 months ago
- Unofficial implementation of GotenNet, new SOTA 3d equivariant transformer, in Pytorchβ67Updated 6 months ago
- Meta Flow Matching: Integrating Vector Fields on the Wasserstein Manifoldβ66Updated 8 months ago
- Erwin: A Tree-based Hierarchical Transformer for Large-scale Physical Systems [ICML'25]β105Updated 3 weeks ago
- β125Updated last year
- [NeurIPS 2025] Official repository for "Energy Matching: Unifying Flow Matching and Energy-Based Models for Generative Modeling"β173Updated 2 weeks ago
- Code repository for Trajectory Flow Matchingβ88Updated last year
- code for "Riemannian Flow Matching on General Geometries".β267Updated last year
- Bare-bones implementations of some generative models in Jax: diffusion, normalizing flows, consistency models, flow matching, (beta)-VAEsβ¦β136Updated last year
- Pytorch implementation of a simple way to enable (Stochastic) Frame Averaging for any networkβ51Updated last year
- Code for the paper https://arxiv.org/abs/2402.04997β97Updated last year
- Improved sampling via learned diffusions (ICLR2024) and an optimal control perspective on diffusion-based generative modeling (TMLR2024)β69Updated 7 months ago
- Implementation of Particle Guidance: non-I.I.D. Diverse Sampling with Diffusion Modelsβ71Updated 2 years ago
- β34Updated 2 months ago
- β39Updated 4 months ago
- Code for the paper Iterated Denoising Energy Matching for Sampling from Boltzmann Densities.β64Updated 6 months ago
- Implementation of the Equiformer, SE3/E3 equivariant attention network that reaches new SOTA, and adopted for use by EquiFold for proteinβ¦β279Updated 10 months ago
- β33Updated last year
- A short and easy PyTorch implementation of E(n) Equivariant Graph Neural Networksβ139Updated 3 years ago
- Lagrangian formulation of Doob's h-transform allowing for efficient rare event samplingβ52Updated 7 months ago
- Generative Flow Networks - GFlowNetβ290Updated this week
- TopoBench is a Python library designed to standardize benchmarking and accelerate research in Topological Deep Learningβ198Updated this week
- Official implementation of Fisher-Flow Matching (NeurIPS 2024).β33Updated last year
- β200Updated last year
- Code for paper E(n)-Equivariant Topological Neural Networksβ36Updated 11 months ago
- Flow Annealed Importance Sampling Bootstrap (FAB). ICLR 2023.β65Updated last year
- Discovering Interpretable Features in Protein Language Models via Sparse Autoencodersβ257Updated this week
- Simple Guidance Mechanisms for Discrete Diffusion Modelsβ57Updated 10 months ago
- Geometric Generative Model Tutorial at LoG 2024β27Updated 11 months ago