├── LICENSE ├── README.md ├── environment.yml ├── momentum ├── __init__.py ├── correlated_topic_models │ ├── __init__.py │ ├── data.py │ ├── dhmc.py │ ├── gibbs.py │ ├── hmc_within_gibbs.py │ ├── mixed_hmc.py │ └── pymc3.py ├── diagnostics │ ├── __init__.py │ ├── convergence.py │ └── ess.py ├── hmc │ ├── __init__.py │ ├── dhmc.py │ ├── hmc_within_gibbs.py │ ├── mixed_hmc_jax.py │ └── naive_mixed_hmc.py ├── potential │ ├── __init__.py │ ├── correlated_topic_models.py │ ├── simple_gmm.py │ └── variable_selection.py ├── simple_gmm │ ├── __init__.py │ ├── dhmc.py │ ├── exact.py │ ├── hmc_within_gibbs.py │ ├── mixed_hmc.py │ ├── nuts.py │ └── pymc3.py ├── utils.py └── variable_selection │ ├── __init__.py │ ├── dhmc.py │ ├── gibbs.py │ ├── hmc_within_gibbs.py │ ├── mixed_hmc.py │ └── pymc3.py ├── scripts ├── correlated_topic_models │ ├── ap_data.joblib │ └── generate_samples_ap_data.py ├── simple_gmm │ ├── generate_samples.py │ ├── generate_samples_24d.py │ ├── test_naive_mixed_hmc.py │ └── test_on_1d_gmm_with_jax.py └── variable_selection │ ├── generate_data.py │ ├── generate_samples.py │ └── simulated_data.joblib └── setup.py /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/StannisZhou/mixed_hmc/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/StannisZhou/mixed_hmc/HEAD/README.md -------------------------------------------------------------------------------- /environment.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/StannisZhou/mixed_hmc/HEAD/environment.yml -------------------------------------------------------------------------------- /momentum/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /momentum/correlated_topic_models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /momentum/correlated_topic_models/data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/StannisZhou/mixed_hmc/HEAD/momentum/correlated_topic_models/data.py -------------------------------------------------------------------------------- /momentum/correlated_topic_models/dhmc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/StannisZhou/mixed_hmc/HEAD/momentum/correlated_topic_models/dhmc.py -------------------------------------------------------------------------------- /momentum/correlated_topic_models/gibbs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/StannisZhou/mixed_hmc/HEAD/momentum/correlated_topic_models/gibbs.py -------------------------------------------------------------------------------- /momentum/correlated_topic_models/hmc_within_gibbs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/StannisZhou/mixed_hmc/HEAD/momentum/correlated_topic_models/hmc_within_gibbs.py -------------------------------------------------------------------------------- /momentum/correlated_topic_models/mixed_hmc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/StannisZhou/mixed_hmc/HEAD/momentum/correlated_topic_models/mixed_hmc.py -------------------------------------------------------------------------------- /momentum/correlated_topic_models/pymc3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/StannisZhou/mixed_hmc/HEAD/momentum/correlated_topic_models/pymc3.py -------------------------------------------------------------------------------- /momentum/diagnostics/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /momentum/diagnostics/convergence.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/StannisZhou/mixed_hmc/HEAD/momentum/diagnostics/convergence.py -------------------------------------------------------------------------------- /momentum/diagnostics/ess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/StannisZhou/mixed_hmc/HEAD/momentum/diagnostics/ess.py -------------------------------------------------------------------------------- /momentum/hmc/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /momentum/hmc/dhmc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/StannisZhou/mixed_hmc/HEAD/momentum/hmc/dhmc.py -------------------------------------------------------------------------------- /momentum/hmc/hmc_within_gibbs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/StannisZhou/mixed_hmc/HEAD/momentum/hmc/hmc_within_gibbs.py -------------------------------------------------------------------------------- /momentum/hmc/mixed_hmc_jax.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/StannisZhou/mixed_hmc/HEAD/momentum/hmc/mixed_hmc_jax.py -------------------------------------------------------------------------------- /momentum/hmc/naive_mixed_hmc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/StannisZhou/mixed_hmc/HEAD/momentum/hmc/naive_mixed_hmc.py -------------------------------------------------------------------------------- /momentum/potential/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /momentum/potential/correlated_topic_models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/StannisZhou/mixed_hmc/HEAD/momentum/potential/correlated_topic_models.py -------------------------------------------------------------------------------- /momentum/potential/simple_gmm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/StannisZhou/mixed_hmc/HEAD/momentum/potential/simple_gmm.py -------------------------------------------------------------------------------- /momentum/potential/variable_selection.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/StannisZhou/mixed_hmc/HEAD/momentum/potential/variable_selection.py -------------------------------------------------------------------------------- /momentum/simple_gmm/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /momentum/simple_gmm/dhmc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/StannisZhou/mixed_hmc/HEAD/momentum/simple_gmm/dhmc.py -------------------------------------------------------------------------------- /momentum/simple_gmm/exact.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/StannisZhou/mixed_hmc/HEAD/momentum/simple_gmm/exact.py -------------------------------------------------------------------------------- /momentum/simple_gmm/hmc_within_gibbs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/StannisZhou/mixed_hmc/HEAD/momentum/simple_gmm/hmc_within_gibbs.py -------------------------------------------------------------------------------- /momentum/simple_gmm/mixed_hmc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/StannisZhou/mixed_hmc/HEAD/momentum/simple_gmm/mixed_hmc.py -------------------------------------------------------------------------------- /momentum/simple_gmm/nuts.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/StannisZhou/mixed_hmc/HEAD/momentum/simple_gmm/nuts.py -------------------------------------------------------------------------------- /momentum/simple_gmm/pymc3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/StannisZhou/mixed_hmc/HEAD/momentum/simple_gmm/pymc3.py -------------------------------------------------------------------------------- /momentum/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/StannisZhou/mixed_hmc/HEAD/momentum/utils.py -------------------------------------------------------------------------------- /momentum/variable_selection/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /momentum/variable_selection/dhmc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/StannisZhou/mixed_hmc/HEAD/momentum/variable_selection/dhmc.py -------------------------------------------------------------------------------- /momentum/variable_selection/gibbs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/StannisZhou/mixed_hmc/HEAD/momentum/variable_selection/gibbs.py -------------------------------------------------------------------------------- /momentum/variable_selection/hmc_within_gibbs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/StannisZhou/mixed_hmc/HEAD/momentum/variable_selection/hmc_within_gibbs.py -------------------------------------------------------------------------------- /momentum/variable_selection/mixed_hmc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/StannisZhou/mixed_hmc/HEAD/momentum/variable_selection/mixed_hmc.py -------------------------------------------------------------------------------- /momentum/variable_selection/pymc3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/StannisZhou/mixed_hmc/HEAD/momentum/variable_selection/pymc3.py -------------------------------------------------------------------------------- /scripts/correlated_topic_models/ap_data.joblib: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/StannisZhou/mixed_hmc/HEAD/scripts/correlated_topic_models/ap_data.joblib -------------------------------------------------------------------------------- /scripts/correlated_topic_models/generate_samples_ap_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/StannisZhou/mixed_hmc/HEAD/scripts/correlated_topic_models/generate_samples_ap_data.py -------------------------------------------------------------------------------- /scripts/simple_gmm/generate_samples.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/StannisZhou/mixed_hmc/HEAD/scripts/simple_gmm/generate_samples.py -------------------------------------------------------------------------------- /scripts/simple_gmm/generate_samples_24d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/StannisZhou/mixed_hmc/HEAD/scripts/simple_gmm/generate_samples_24d.py -------------------------------------------------------------------------------- /scripts/simple_gmm/test_naive_mixed_hmc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/StannisZhou/mixed_hmc/HEAD/scripts/simple_gmm/test_naive_mixed_hmc.py -------------------------------------------------------------------------------- /scripts/simple_gmm/test_on_1d_gmm_with_jax.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/StannisZhou/mixed_hmc/HEAD/scripts/simple_gmm/test_on_1d_gmm_with_jax.py -------------------------------------------------------------------------------- /scripts/variable_selection/generate_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/StannisZhou/mixed_hmc/HEAD/scripts/variable_selection/generate_data.py -------------------------------------------------------------------------------- /scripts/variable_selection/generate_samples.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/StannisZhou/mixed_hmc/HEAD/scripts/variable_selection/generate_samples.py -------------------------------------------------------------------------------- /scripts/variable_selection/simulated_data.joblib: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/StannisZhou/mixed_hmc/HEAD/scripts/variable_selection/simulated_data.joblib -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/StannisZhou/mixed_hmc/HEAD/setup.py --------------------------------------------------------------------------------