Illustration drawn by RepoMasters from the repo’s ID and its language color. Credits
PrototypeCommon

soran-ghaderi/torchebm

🍓 Simulation-free, GPU-first generative modeling in PyTorch ⚡ Composable primitives for scalable, stable training of modern EBMs, diffusion, flow matching, and Schrödinger bridges.

Owner
soran-ghaderi (personal account)
Stars
Stars: 117
Forks
Forks: 24
Language
Python
License
MIT
Created
2024
Topics
contrastive-divergencecudalangevin-dynamicsprobabilistic-machine-learningsampling-methodsscore-matchingdiffusion-modelshamiltonian-monte-carloequilibrium-matchingenergy-based-models
Tier
Prototype (100 to 249 ★) What this tier means
View on GitHub

Nearby in Python

All Python repos