This repository contains IPython notebooks for running selected experiments from Reverse Engineering the Neural Tangent Kernel. Notebooks are all compatible with Google Colab.

This library makes heavy use of neural_tangents, a Google library for fast net kernel computations, and eigenlearning, a library from Jamie Simon that provides functions for synthetic data generation and network training with neural_tangents. Everything here runs as of 2022 but might not be maintained.

Direct any questions to [email protected]!