Skip to content
TorchEBM
datasets
Initializing search
soran-ghaderi/torchebm
Home
Getting Started
API Reference
Examples
Developer Guide
Benchmarks
Blog
FAQ
TorchEBM
soran-ghaderi/torchebm
Home
Getting Started
Getting Started
Concepts
Concepts
Design and Scope
The Energy-Based View
Sampling and Integration
Learning Objectives
Interpolants and Couplings
API Reference
API Reference
torchebm
torchebm
core
core
base_coupling
base_integrator
base_interpolant
base_loss
base_model
base_module
base_sampler
base_scheduler
base_trainer
schedulable
couplings
couplings
coupling_utils
independent
model_induced
ot
cuda
cuda
fused_langevin
datasets
datasets
generators
distributed
integrators
integrators
adaptive_heun
bosh3
dopri
euler_maruyama
heun
integrator_utils
leapfrog
midpoint
rk4
interpolants
interpolants
cosine
interpolant_utils
linear
variance_preserving
losses
losses
contrastive_divergence
energy_matching
equilibrium_matching
loss_utils
score_matching
models
models
components
components
embeddings
heads
patch
positional
transformer
conditional_transformer_2d
wrappers
samplers
samplers
flow
gradient_descent
hmc
langevin_dynamics
utils
utils
profiling
training
Examples
Examples
Foundations
Foundations
Energy
Energy
Energy Landscapes
Custom Energies
Datasets
Datasets
Dataset Gallery
Schedulers
Schedulers
Scheduler Anatomy
Interpolants
Interpolants
Interpolant Anatomy
Sampling
Sampling
MCMC
MCMC
Langevin Dynamics 101
HMC 101
Parallel Chains
Integrators
Integrators
Integrator Comparison
Flow
Flow
FlowSampler ODE 101
Training
Training
MCMC Losses
MCMC Losses
CD-k on Two Moons
Persistent CD
Score Matching
Score Matching
Denoising Score Matching
Equilibrium Matching
Equilibrium Matching
Equilibrium Matching in 2D
Energy Matching
Energy Matching
Energy Matching in 2D
Energy Matching: the paper's 2D suite
Couplings
Couplings
Coupling Comparison
Showcase
Showcase
Density Transport
Integrators vs. the Exact Solution
Integrators in Practice: an Interactive Explainer
Developer Guide
Developer Guide
Architecture
Code Guidelines
Performance and Benchmarking
Benchmarks
Blog
FAQ
Home
API Reference
torchebm
datasets
torchebm.datasets
¶
Back to top