AnandShegde/bnn-zoo

bnn-zoo

This repository contains implementation of MLP(Multi-layer Perceptron) from scratch using JAX, scikit-learn, matplotlib, seaborn, probml-utils for latexifying and jaxifying.

Contents:

1. Basic MLP
2. Gaussian MLP
3. Deep Ensemble with and without Gaussian MLP
4. Bootstrap with and without Gaussian MLP
5. MC Dropout

For both Classification and Regression.