soran-ghaderi/torchebm
🍓 Build and train energy-based and diffusion models in PyTorch ⚡.
Provides composable training objectives (contrastive divergence, score matching, equilibrium matching) and multiple sampling strategies—MCMC, optimization, and learned continuous-time dynamics via ODE/SDE integration. Includes pre-built analytical potentials, neural network architectures for conditional generation, synthetic benchmarking datasets, and numerical integrators for Hamiltonian dynamics. Targets PyTorch with CUDA acceleration and mixed precision support for efficient energy landscape exploration.
Available on PyPI.
Stars
81
Forks
7
Language
Python
License
MIT
Category
Last pushed
Feb 17, 2026
Commits (30d)
0
Dependencies
3
Get this data via API
curl "https://pt-edge.onrender.com/api/v1/quality/generative-ai/soran-ghaderi/torchebm"
Open to everyone — 100 requests/day, no key needed. Get a free key for 1,000/day.
Related tools
opendilab/GenerativeRL
Python library for solving reinforcement learning (RL) problems using generative models (e.g....
spqb/adabmDCApy
Pytorch implementation of adabmDCA
kidist-amde/ddro
We introduce the direct document relevance optimization (DDRO) for training a pairwise ranker...
AstraZeneca/DiffAbXL
The official implementation of DiffAbXL benchmarked in the paper "Exploring Log-Likelihood...
G-U-N/UniRL
a unified reinforcement learning toolbox for joint RL on language models and diffusion models