.gitignore
CHANGELOG.md
CITATION.cff
CODE_OF_CONDUCT.md
CONTRIBUTING.md
LICENSE
MANIFEST.in
README.md
cliff.toml
generate_api_docs.py
pyproject.toml
requirements-docs.txt
requirements.txt
torchebm/__init__.py
torchebm/_version.py
torchebm.egg-info/PKG-INFO
torchebm.egg-info/SOURCES.txt
torchebm.egg-info/dependency_links.txt
torchebm.egg-info/requires.txt
torchebm.egg-info/scm_file_list.json
torchebm.egg-info/scm_version.json
torchebm.egg-info/top_level.txt
torchebm/core/__init__.py
torchebm/core/base_coupling.py
torchebm/core/base_integrator.py
torchebm/core/base_interpolant.py
torchebm/core/base_loss.py
torchebm/core/base_model.py
torchebm/core/base_module.py
torchebm/core/base_sampler.py
torchebm/core/base_scheduler.py
torchebm/core/base_trainer.py
torchebm/core/schedulable.py
torchebm/couplings/__init__.py
torchebm/couplings/coupling_utils.py
torchebm/couplings/independent.py
torchebm/couplings/model_induced.py
torchebm/couplings/ot.py
torchebm/cuda/__init__.py
torchebm/cuda/fused_langevin.py
torchebm/datasets/__init__.py
torchebm/datasets/generators.py
torchebm/integrators/__init__.py
torchebm/integrators/adaptive_heun.py
torchebm/integrators/bosh3.py
torchebm/integrators/dopri.py
torchebm/integrators/euler_maruyama.py
torchebm/integrators/heun.py
torchebm/integrators/integrator_utils.py
torchebm/integrators/leapfrog.py
torchebm/integrators/midpoint.py
torchebm/integrators/rk4.py
torchebm/interpolants/__init__.py
torchebm/interpolants/cosine.py
torchebm/interpolants/interpolant_utils.py
torchebm/interpolants/linear.py
torchebm/interpolants/variance_preserving.py
torchebm/losses/__init__.py
torchebm/losses/contrastive_divergence.py
torchebm/losses/energy_matching.py
torchebm/losses/equilibrium_matching.py
torchebm/losses/loss_utils.py
torchebm/losses/score_matching.py
torchebm/models/__init__.py
torchebm/models/conditional_transformer_2d.py
torchebm/models/wrappers.py
torchebm/models/components/__init__.py
torchebm/models/components/embeddings.py
torchebm/models/components/heads.py
torchebm/models/components/patch.py
torchebm/models/components/positional.py
torchebm/models/components/transformer.py
torchebm/samplers/__init__.py
torchebm/samplers/flow.py
torchebm/samplers/gradient_descent.py
torchebm/samplers/hmc.py
torchebm/samplers/langevin_dynamics.py
torchebm/utils/__init__.py
torchebm/utils/distributed.py
torchebm/utils/profiling.py
torchebm/utils/training.py