Repository, bayesian-ntk, to accompany the paper Bayesian Deep Ensembles via the Neural Tangent Kernel and our paper Building Bayesian Neural Networks with blocks: On Structure, Interpretability and Uncertainty. Please note that analytic NTKGP posterior is now implemented in neural-tangents, more details here.

To install requirements:

pip install -r requirements.txt

The functionality of bayesian-ntk and the application on our Bayesian Additive Neural Network is explained in our notebook bann-ntk_toy_example.ipynb, which you can here.