Jax samplers: 3 ways

blog post: link

Code to reproduce the results in the "Jax 3 ways" blog post.

Installation:

To run all the examples, simply run the bash script: run_all_samplers.sh

Name		Name	Last commit message	Last commit date
outputs		outputs
.gitignore		.gitignore
README.md		README.md
logistic_regression_model.py		logistic_regression_model.py
mala_samplers.py		mala_samplers.py
requirements.txt		requirements.txt
run_MALA_increase_data.py		run_MALA_increase_data.py
run_MALA_increase_dimension.py		run_MALA_increase_dimension.py
run_SGLD_increase_data.py		run_SGLD_increase_data.py
run_SGLD_increase_dimension.py		run_SGLD_increase_dimension.py
run_ULA_increase_data.py		run_ULA_increase_data.py
run_ULA_increase_dimension.py		run_ULA_increase_dimension.py
run_all_samplers.sh		run_all_samplers.sh
run_python_sampler.py		run_python_sampler.py
sgld_samplers.py		sgld_samplers.py
ula_samplers.py		ula_samplers.py