activatedgeek/svgd

PyTorch implementation of Stein Variational Gradient Descent

The Stein Gradient

This is the complementary notebook repository for the blog post The Stein Gradient at https://sanyamkapoor.com/kb/the-stein-gradient