meta-inf/probabilistic-regression

A short tutorial on Bayesian Neural Networks with JAX

probabilistic-regression

An implementation of Bayes-by-Backprop (Blundell et al. 2015).

Based on this tutorial by tensorflow probability, but implemented with JAX.