.gitignore
LICENSE
README.md
__init__.py
generate_api_docs.py
mkdocs.yml
pyproject.toml
requirements-docs.txt
requirements.txt
.github/workflows/docs_ci.yml
.github/workflows/publish.yml
.github/workflows/tag-release.yml
docs/faq.md
docs/index.md
docs/tags.md
docs/api/index.md
docs/api/torchebm/index.md
docs/api/torchebm/core/index.md
docs/api/torchebm/core/base_energy_function/index.md
docs/api/torchebm/core/base_energy_function/classes/AckleyEnergy.md
docs/api/torchebm/core/base_energy_function/classes/BaseEnergyFunction.md
docs/api/torchebm/core/base_energy_function/classes/DoubleWellEnergy.md
docs/api/torchebm/core/base_energy_function/classes/GaussianEnergy.md
docs/api/torchebm/core/base_energy_function/classes/HarmonicEnergy.md
docs/api/torchebm/core/base_energy_function/classes/RastriginEnergy.md
docs/api/torchebm/core/base_energy_function/classes/RosenbrockEnergy.md
docs/api/torchebm/core/base_loss/index.md
docs/api/torchebm/core/base_loss/classes/BaseContrastiveDivergence.md
docs/api/torchebm/core/base_loss/classes/BaseLoss.md
docs/api/torchebm/core/base_optimizer/index.md
docs/api/torchebm/core/base_optimizer/classes/Optimizer.md
docs/api/torchebm/core/base_sampler/index.md
docs/api/torchebm/core/base_sampler/classes/BaseSampler.md
docs/api/torchebm/core/base_trainer/index.md
docs/api/torchebm/core/base_trainer/classes/ContrastiveDivergenceTrainer.md
docs/api/torchebm/cuda/index.md
docs/api/torchebm/datasets/index.md
docs/api/torchebm/datasets/generators/index.md
docs/api/torchebm/datasets/generators/classes/BaseSyntheticDataset.md
docs/api/torchebm/datasets/generators/classes/CheckerboardDataset.md
docs/api/torchebm/datasets/generators/classes/CircleDataset.md
docs/api/torchebm/datasets/generators/classes/EightGaussiansDataset.md
docs/api/torchebm/datasets/generators/classes/GaussianMixtureDataset.md
docs/api/torchebm/datasets/generators/classes/GridDataset.md
docs/api/torchebm/datasets/generators/classes/PinwheelDataset.md
docs/api/torchebm/datasets/generators/classes/SwissRollDataset.md
docs/api/torchebm/datasets/generators/classes/TwoMoonsDataset.md
docs/api/torchebm/losses/index.md
docs/api/torchebm/losses/contrastive_divergence/index.md
docs/api/torchebm/losses/contrastive_divergence/classes/ContrastiveDivergence.md
docs/api/torchebm/losses/contrastive_divergence/classes/ParallelTemperingCD.md
docs/api/torchebm/losses/contrastive_divergence/classes/PersistentContrastiveDivergence.md
docs/api/torchebm/losses/score_matching/index.md
docs/api/torchebm/losses/score_matching/classes/ScoreMatching.md
docs/api/torchebm/models/index.md
docs/api/torchebm/models/base_model/index.md
docs/api/torchebm/models/base_model/classes/BaseModel.md
docs/api/torchebm/samplers/index.md
docs/api/torchebm/samplers/hmc/index.md
docs/api/torchebm/samplers/hmc/classes/HamiltonianMonteCarlo.md
docs/api/torchebm/samplers/langevin_dynamics/index.md
docs/api/torchebm/samplers/langevin_dynamics/classes/LangevinDynamics.md
docs/api/torchebm/utils/index.md
docs/api/torchebm/utils/visualization/index.md
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/visualization/advanced_energy_landscape.png
docs/assets/images/visualization/basic_energy_landscape.png
docs/assets/images/visualization/distribution_comparison.png
docs/assets/images/visualization/energy_evolution.png
docs/assets/images/visualization/interactive_visualization.png
docs/assets/images/visualization/langevin_trajectory.png
docs/assets/images/visualization/multiple_chains.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/api_design.md
docs/developer_guide/api_generation.md
docs/developer_guide/architecture.md
docs/developer_guide/code_style.md
docs/developer_guide/commit_conventions.md
docs/developer_guide/contributing.md
docs/developer_guide/core_components.md
docs/developer_guide/cuda_optimizations.md
docs/developer_guide/design_principles.md
docs/developer_guide/development_setup.md
docs/developer_guide/implementation_energy.md
docs/developer_guide/implementation_losses.md
docs/developer_guide/implementation_models.md
docs/developer_guide/implementation_samplers.md
docs/developer_guide/index.md
docs/developer_guide/performance.md
docs/developer_guide/project_structure.md
docs/developer_guide/testing_guide.md
docs/examples/index.md
docs/examples/datasets/index.md
docs/examples/energy_functions/index.md
docs/examples/samplers/hmc.md
docs/examples/samplers/langevin_dynamics.md
docs/examples/samplers/langevin_trajectory.md
docs/examples/training/training_ebm_gaussian.md
docs/examples/visualization/energy_visualization.md
docs/examples/visualization/index.md
docs/guides/custom_neural_networks.md
docs/guides/energy_functions.md
docs/guides/getting_started.md
docs/guides/index.md
docs/guides/introduction.md
docs/guides/loss_functions.md
docs/guides/parallel_sampling.md
docs/guides/samplers.md
docs/guides/training.md
docs/guides/visualization.md
docs/hooks/socialmedia.py
docs/overrides/partials/outdated.html
docs/stylesheets/extra.css
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/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_energy_function.py
tests/core/test_loss.py
tests/datasets/test_generators.py
tests/losses/test_contrastive_divergence.py
tests/samplers/test_hmc.py
tests/samplers/test_langevin_dynamics.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_energy_function.py
torchebm/core/base_loss.py
torchebm/core/base_optimizer.py
torchebm/core/base_sampler.py
torchebm/core/base_trainer.py
torchebm/cuda/__init__.py
torchebm/datasets/__init__.py
torchebm/datasets/generators.py
torchebm/losses/__init__.py
torchebm/losses/contrastive_divergence.py
torchebm/losses/score_matching.py
torchebm/models/__init__.py
torchebm/models/base_model.py
torchebm/samplers/__init__.py
torchebm/samplers/hmc.py
torchebm/samplers/langevin_dynamics.py
torchebm/utils/__init__.py
torchebm/utils/visualization.py