Code for the paper: "Tensor Programs II: Neural Tangent Kernel for Any Architecture"
Primary LanguageJupyter Notebook