/nngp-ntk-simple

Code demonstrating how to compute NNGP and NT kernels for simple neural network architectures.

Primary LanguageJupyter Notebook

nngp-ntk-simple

Code demonstrating how to compute NNGP and NT kernels for simple neural network architectures.