.gitignore
CHANGELOG.md
CODE_OF_CONDUCT.md
LICENSE
README.md
__init__.py
generate_api_docs.py
mkdocs.yml
pyproject.toml
requirements-docs.txt
requirements.txt
.github/dependabot.yml
.github/ISSUE_TEMPLATE/bug_report.yml
.github/ISSUE_TEMPLATE/config.yml
.github/ISSUE_TEMPLATE/feature_request.yml
.github/workflows/ci.yml
.github/workflows/docs_ci.yml
.github/workflows/publish.yml
.github/workflows/tag-release.yml
.github/workflows/update-changelog.yml
docs/faq.md
docs/index.md
docs/tags.md
docs/api/index.md
docs/assets/animations/ebm_training_animation.gif
docs/assets/images/energy_grid_bg.png
docs/assets/images/favicon.svg
docs/assets/images/logo_with_text.svg
docs/assets/images/nabla_icon.svg
docs/assets/images/parallel.jpg
docs/assets/images/sampling.jpg
docs/assets/images/trajectory.jpg
docs/assets/images/blog/hamiltonian/non-L-pendulum.png
docs/assets/images/blog/hamiltonian/numerical_err.png
docs/assets/images/blog/hamiltonian/pendulum_phase_space.mp4
docs/assets/images/blog/hamiltonian/quadratics.png
docs/assets/images/datasets/checkerboard.png
docs/assets/images/datasets/circle.png
docs/assets/images/datasets/eight_gaussians.png
docs/assets/images/datasets/gaussian_mixture.png
docs/assets/images/datasets/grid.png
docs/assets/images/datasets/pinwheel.png
docs/assets/images/datasets/swiss_roll.png
docs/assets/images/datasets/two_moons.png
docs/assets/images/e_functions/ackley.png
docs/assets/images/e_functions/double_well.png
docs/assets/images/e_functions/gaussian.png
docs/assets/images/e_functions/harmonic.png
docs/assets/images/e_functions/rastrigin.png
docs/assets/images/e_functions/rosenbrock.png
docs/assets/images/examples/double_well_trajectory.png
docs/assets/images/examples/energy_landscape_epoch_10.png
docs/assets/images/examples/energy_landscape_epoch_100.png
docs/assets/images/examples/energy_landscape_epoch_20.png
docs/assets/images/examples/energy_landscape_epoch_30.png
docs/assets/images/examples/energy_landscape_epoch_40.png
docs/assets/images/examples/energy_landscape_epoch_50.png
docs/assets/images/examples/energy_landscape_epoch_60.png
docs/assets/images/examples/energy_landscape_epoch_70.png
docs/assets/images/examples/energy_landscape_epoch_80.png
docs/assets/images/examples/energy_landscape_epoch_90.png
docs/assets/images/examples/energy_landscape_final.png
docs/assets/images/examples/gaussian_mixture_target.png
docs/assets/images/examples/hmc_basic.png
docs/assets/images/examples/hmc_comparison.png
docs/assets/images/examples/hmc_custom_mass.png
docs/assets/images/examples/hmc_standard.png
docs/assets/images/examples/langevin_basic.png
docs/assets/images/examples/langevin_trajectory.png
docs/assets/images/loss_functions/cd_training_loss.png
docs/assets/images/samplers/langevin_trajectory.png
docs/assets/images/samplers/multiple_chains.png
docs/assets/images/training/cd_training_loss.png
docs/assets/images/training/ebm_training_epoch_1.png
docs/assets/images/training/ebm_training_epoch_100.png
docs/assets/images/training/ebm_training_epoch_50.png
docs/assets/images/training/energy_evolution.png
docs/assets/images/training/loss_comparison.png
docs/assets/images/visualization/basic_energy_landscape.png
docs/assets/images/visualization/distribution_comparison_updated.png
docs/assets/images/visualization/energy_evolution.png
docs/assets/images/visualization/energy_evolution_updated.png
docs/assets/images/visualization/langevin_trajectory.png
docs/assets/images/visualization/langevin_trajectory_updated.png
docs/assets/images/visualization/loss_comparison.png
docs/assets/images/visualization/multiple_chains.png
docs/assets/images/visualization/multiple_chains_updated.png
docs/assets/images/visualization/parameter_comparison.png
docs/assets/images/visualization/probability_density.png
docs/assets/images/visualization/training_loss.png
docs/blog/.authors.yml
docs/blog/index.md
docs/blog/posts/code-examples.md
docs/blog/posts/hamiltonian-mechanics.md
docs/blog/posts/langevin_dynamics_sampling.md
docs/blog/posts/neural_energy_functions.md
docs/blog/posts/schrodinger-bridges.md
docs/developer_guide/architecture.md
docs/developer_guide/code_guidelines.md
docs/developer_guide/getting_started.md
docs/developer_guide/index.md
docs/developer_guide/performance.md
docs/examples/index.md
docs/examples/datasets/index.md
docs/examples/models/index.md
docs/examples/samplers/index.md
docs/examples/training/index.md
docs/examples/visualization/index.md
docs/hooks/socialmedia.py
docs/javascripts/cards-clickable.js
docs/overrides/partials/outdated.html
docs/stylesheets/extra.css
docs/tutorials/custom_neural_networks.md
docs/tutorials/getting_started.md
docs/tutorials/index.md
docs/tutorials/loss_functions.md
docs/tutorials/parallel_sampling.md
docs/tutorials/samplers.md
docs/tutorials/training.md
docs/tutorials/visualization.md
examples/README.md
examples/main.py
examples/core/energy_advanced.py
examples/core/energy_basic.py
examples/core/energy_functions/landscape_2d.py
examples/core/energy_functions/multimodal.py
examples/core/energy_functions/parametric.py
examples/datasets/generate_dataset_imgs.py
examples/samplers/hmc/advanced.py
examples/samplers/hmc/gaussian_sampling.py
examples/samplers/hmc/mass_matrix.py
examples/samplers/langevin/advanced.py
examples/samplers/langevin/gaussian_sampling.py
examples/samplers/langevin/multimodal_sampling.py
examples/samplers/langevin/visualization_trajectory.py
examples/training_models/mlp_cd_training.py
examples/training_models/mlp_sm_training.py
examples/visualization/samplers_vis.py
examples/visualization/utils.py
examples/visualization/advanced/energy_evolution.py
examples/visualization/advanced/energy_landscape.py
examples/visualization/advanced/energy_over_time.py
examples/visualization/advanced/langevin_trajectory.py
examples/visualization/advanced/multiple_chains.py
examples/visualization/advanced/multiple_doublewell.py
examples/visualization/advanced/parallel_chains.py
examples/visualization/advanced/trajectory_animation.py
examples/visualization/advanced/visualization_examples.py
examples/visualization/basic/contour_plots.py
examples/visualization/basic/current_energy_fns_vis.py
examples/visualization/basic/distribution_comparison.py
examples/visualization/basic/energy_fn_vis.py
examples/visualization/basic/energy_landscape.py
tests/conftest.py
tests/core/test_base_scheduler.py
tests/core/test_energy_function.py
tests/core/test_loss.py
tests/datasets/test_generators.py
tests/integrators/test_euler_maruyama.py
tests/integrators/test_heun.py
tests/integrators/test_integrator_utils.py
tests/integrators/test_leapfrog.py
tests/interpolants/__init__.py
tests/interpolants/test_base_interpolant.py
tests/interpolants/test_cosine.py
tests/interpolants/test_linear.py
tests/interpolants/test_variance_preserving.py
tests/losses/test_contrastive_divergence.py
tests/losses/test_denoising_score_matching.py
tests/losses/test_equilibrium_matching.py
tests/losses/test_exact_score_matching.py
tests/losses/test_loss_utils.py
tests/losses/test_sliced_score_matching.py
tests/samplers/test_flow.py
tests/samplers/test_gradient_descent.py
tests/samplers/test_hmc.py
tests/samplers/test_langevin_dynamics.py
tests/utils/test_eqm_utils.py
tests/utils/test_training.py
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/top_level.txt
torchebm/core/__init__.py
torchebm/core/base_integrator.py
torchebm/core/base_interpolant.py
torchebm/core/base_loss.py
torchebm/core/base_model.py
torchebm/core/base_sampler.py
torchebm/core/base_scheduler.py
torchebm/core/base_trainer.py
torchebm/core/device_mixin.py
torchebm/cuda/__init__.py
torchebm/datasets/__init__.py
torchebm/datasets/generators.py
torchebm/integrators/__init__.py
torchebm/integrators/euler_maruyama.py
torchebm/integrators/heun.py
torchebm/integrators/integrator_utils.py
torchebm/integrators/leapfrog.py
torchebm/interpolants/__init__.py
torchebm/interpolants/cosine.py
torchebm/interpolants/linear.py
torchebm/interpolants/variance_preserving.py
torchebm/losses/__init__.py
torchebm/losses/contrastive_divergence.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/image.py
torchebm/utils/training.py
torchebm/utils/visualization.py