Repository, bayesian-ntk, to accompany the paper Bayesian Deep Ensembles via the Neural Tangent Kernel. 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 is explained in our notebook toy_1d_example.ipynb, which you can 