Simple numpy implementation of (Bayesian) Neural Network