Code for "Optimising Distributions with Natural Gradient Surrogates" - View it on GitHub