soran-ghaderi/torchebm
🍓 Build and train energy-based and diffusion models in PyTorch ⚡.
This tool helps researchers and machine learning engineers develop and train generative models. It takes in structured or unstructured data and outputs new, synthetic data samples that resemble the original. Users can define custom energy functions and train models using a variety of objectives and sampling techniques.
Available on PyPI.
Use this if you are a machine learning researcher or engineer exploring advanced generative modeling techniques, particularly those based on energy functions, diffusion, or flow models.
Not ideal if you need a simple, off-the-shelf solution for basic data generation or if you are not familiar with PyTorch and advanced deep learning concepts.
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
spqb/adabmDCApy
Pytorch implementation of adabmDCA
opendilab/GenerativeRL
Python library for solving reinforcement learning (RL) problems using generative models (e.g....
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
kidist-amde/ddro
We introduce the direct document relevance optimization (DDRO) for training a pairwise ranker...