├── .DS_Store ├── .github ├── copilot-instructions.md └── workflows │ ├── quick-test.yml │ └── test.yml ├── .gitignore ├── .pre-commit-config.yaml ├── .vscode └── settings.json ├── LICENSE.md ├── README.md ├── data ├── MCMC_SyntheticData.py └── plots │ ├── EpistemicAleatoricUncertainty.png │ ├── GMM_HMC1.gif │ ├── HMC_Sampler.gif │ ├── HMC_Sampler2.gif │ ├── HMC_Sampler3.gif │ └── HMC_Sampler4.gif ├── experiments ├── .DS_Store ├── AIS.py ├── AIS_energy_viz.py ├── AIS_old.py ├── AIs_viz.py ├── HMC.py ├── HMC_2D.py ├── MALA.py ├── MH_NeuralNetwork.py ├── MH_NeuralNetwork2.py ├── MH_NeuralNetwork3.py ├── MH_sampler.py ├── MH_sampler2.py ├── MH_sampler2D.py ├── ais_energy_evolution.gif ├── functional_pytorch ├── functional_pytorch.py ├── langevin_sampler.py └── various.py ├── pyproject.toml ├── src └── mcmc │ ├── __init__.py │ ├── chain.py │ ├── data.py │ ├── energy.py │ ├── sampler.py │ └── utils.py ├── tests ├── conftest.py ├── test_energy.py ├── test_functionality.py ├── test_sampler.py ├── test_tensordict_key_order.py └── test_utils.py └── uv.lock /.DS_Store: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ludwigwinkler/pytorch_MCMC/HEAD/.DS_Store -------------------------------------------------------------------------------- /.github/copilot-instructions.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ludwigwinkler/pytorch_MCMC/HEAD/.github/copilot-instructions.md -------------------------------------------------------------------------------- /.github/workflows/quick-test.yml: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /.github/workflows/test.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ludwigwinkler/pytorch_MCMC/HEAD/.github/workflows/test.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ludwigwinkler/pytorch_MCMC/HEAD/.gitignore -------------------------------------------------------------------------------- /.pre-commit-config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ludwigwinkler/pytorch_MCMC/HEAD/.pre-commit-config.yaml -------------------------------------------------------------------------------- /.vscode/settings.json: -------------------------------------------------------------------------------- 1 | { 2 | } -------------------------------------------------------------------------------- /LICENSE.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ludwigwinkler/pytorch_MCMC/HEAD/LICENSE.md -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ludwigwinkler/pytorch_MCMC/HEAD/README.md -------------------------------------------------------------------------------- /data/MCMC_SyntheticData.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ludwigwinkler/pytorch_MCMC/HEAD/data/MCMC_SyntheticData.py -------------------------------------------------------------------------------- /data/plots/EpistemicAleatoricUncertainty.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ludwigwinkler/pytorch_MCMC/HEAD/data/plots/EpistemicAleatoricUncertainty.png -------------------------------------------------------------------------------- /data/plots/GMM_HMC1.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ludwigwinkler/pytorch_MCMC/HEAD/data/plots/GMM_HMC1.gif -------------------------------------------------------------------------------- /data/plots/HMC_Sampler.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ludwigwinkler/pytorch_MCMC/HEAD/data/plots/HMC_Sampler.gif -------------------------------------------------------------------------------- /data/plots/HMC_Sampler2.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ludwigwinkler/pytorch_MCMC/HEAD/data/plots/HMC_Sampler2.gif -------------------------------------------------------------------------------- /data/plots/HMC_Sampler3.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ludwigwinkler/pytorch_MCMC/HEAD/data/plots/HMC_Sampler3.gif -------------------------------------------------------------------------------- /data/plots/HMC_Sampler4.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ludwigwinkler/pytorch_MCMC/HEAD/data/plots/HMC_Sampler4.gif -------------------------------------------------------------------------------- /experiments/.DS_Store: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ludwigwinkler/pytorch_MCMC/HEAD/experiments/.DS_Store -------------------------------------------------------------------------------- /experiments/AIS.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ludwigwinkler/pytorch_MCMC/HEAD/experiments/AIS.py -------------------------------------------------------------------------------- /experiments/AIS_energy_viz.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ludwigwinkler/pytorch_MCMC/HEAD/experiments/AIS_energy_viz.py -------------------------------------------------------------------------------- /experiments/AIS_old.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ludwigwinkler/pytorch_MCMC/HEAD/experiments/AIS_old.py -------------------------------------------------------------------------------- /experiments/AIs_viz.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ludwigwinkler/pytorch_MCMC/HEAD/experiments/AIs_viz.py -------------------------------------------------------------------------------- /experiments/HMC.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ludwigwinkler/pytorch_MCMC/HEAD/experiments/HMC.py -------------------------------------------------------------------------------- /experiments/HMC_2D.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ludwigwinkler/pytorch_MCMC/HEAD/experiments/HMC_2D.py -------------------------------------------------------------------------------- /experiments/MALA.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ludwigwinkler/pytorch_MCMC/HEAD/experiments/MALA.py -------------------------------------------------------------------------------- /experiments/MH_NeuralNetwork.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ludwigwinkler/pytorch_MCMC/HEAD/experiments/MH_NeuralNetwork.py -------------------------------------------------------------------------------- /experiments/MH_NeuralNetwork2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ludwigwinkler/pytorch_MCMC/HEAD/experiments/MH_NeuralNetwork2.py -------------------------------------------------------------------------------- /experiments/MH_NeuralNetwork3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ludwigwinkler/pytorch_MCMC/HEAD/experiments/MH_NeuralNetwork3.py -------------------------------------------------------------------------------- /experiments/MH_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ludwigwinkler/pytorch_MCMC/HEAD/experiments/MH_sampler.py -------------------------------------------------------------------------------- /experiments/MH_sampler2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ludwigwinkler/pytorch_MCMC/HEAD/experiments/MH_sampler2.py -------------------------------------------------------------------------------- /experiments/MH_sampler2D.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ludwigwinkler/pytorch_MCMC/HEAD/experiments/MH_sampler2D.py -------------------------------------------------------------------------------- /experiments/ais_energy_evolution.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ludwigwinkler/pytorch_MCMC/HEAD/experiments/ais_energy_evolution.gif -------------------------------------------------------------------------------- /experiments/functional_pytorch: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ludwigwinkler/pytorch_MCMC/HEAD/experiments/functional_pytorch -------------------------------------------------------------------------------- /experiments/functional_pytorch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ludwigwinkler/pytorch_MCMC/HEAD/experiments/functional_pytorch.py -------------------------------------------------------------------------------- /experiments/langevin_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ludwigwinkler/pytorch_MCMC/HEAD/experiments/langevin_sampler.py -------------------------------------------------------------------------------- /experiments/various.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ludwigwinkler/pytorch_MCMC/HEAD/experiments/various.py -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ludwigwinkler/pytorch_MCMC/HEAD/pyproject.toml -------------------------------------------------------------------------------- /src/mcmc/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ludwigwinkler/pytorch_MCMC/HEAD/src/mcmc/__init__.py -------------------------------------------------------------------------------- /src/mcmc/chain.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/mcmc/data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ludwigwinkler/pytorch_MCMC/HEAD/src/mcmc/data.py -------------------------------------------------------------------------------- /src/mcmc/energy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ludwigwinkler/pytorch_MCMC/HEAD/src/mcmc/energy.py -------------------------------------------------------------------------------- /src/mcmc/sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ludwigwinkler/pytorch_MCMC/HEAD/src/mcmc/sampler.py -------------------------------------------------------------------------------- /src/mcmc/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ludwigwinkler/pytorch_MCMC/HEAD/src/mcmc/utils.py -------------------------------------------------------------------------------- /tests/conftest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ludwigwinkler/pytorch_MCMC/HEAD/tests/conftest.py -------------------------------------------------------------------------------- /tests/test_energy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ludwigwinkler/pytorch_MCMC/HEAD/tests/test_energy.py -------------------------------------------------------------------------------- /tests/test_functionality.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ludwigwinkler/pytorch_MCMC/HEAD/tests/test_functionality.py -------------------------------------------------------------------------------- /tests/test_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ludwigwinkler/pytorch_MCMC/HEAD/tests/test_sampler.py -------------------------------------------------------------------------------- /tests/test_tensordict_key_order.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/test_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ludwigwinkler/pytorch_MCMC/HEAD/tests/test_utils.py -------------------------------------------------------------------------------- /uv.lock: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ludwigwinkler/pytorch_MCMC/HEAD/uv.lock --------------------------------------------------------------------------------