probml / pyprobml

Python code for "Probabilistic Machine learning" book by Kevin Murphy
MIT License
6.53k stars 1.53k forks source link

Added `bnn_mnist_sgld.ipynb` #987

Closed karm-patel closed 2 years ago

karm-patel commented 2 years ago

Description

Added flax & blackjax implementation of SGLD.

Figure Number

Fig 19.8, 19.9

Figures

After: image

After: image

Issue

856

Checklist

review-notebook-app[bot] commented 2 years ago

Check out this pull request on  ReviewNB

See visual diffs & provide feedback on Jupyter Notebooks.


Powered by ReviewNB

karm-patel commented 2 years ago

Dr. @murphyk, You can ignore 3rd workflow check, since it is failing due to time out.

murphyk commented 2 years ago

Excellent work!