blog post: link
Code to reproduce the results in the "Jax 3 ways" blog post.
virtualenv venv; source venv/bin/activate
pip install -r requirements.txt
To run all the examples, simply run the bash script: run_all_samplers.sh
run_ULA_increase_data.py
run_ULA_increase_dimension.py
run_MALA_increase_data.py
run_MALA_increase_dimension.py
run_SGLD_increase_data.py
run_SGLD_increase_dimension.py