soran-ghaderi/torchebm
🍓 Build and train energy-based and diffusion models in PyTorch ⚡.
GitHub repository with 92 stars and 10 forks.
Language: Python
Topics: contrastive-divergence, cuda, diffusion-models, energy-based-model, equilibrium, equilibrium-matching, equilibrium-modeling, generative-ai, hamiltonian, hamiltonian-monte-carlo