TorchNTK: A Library for Calculation of Neural Tangent Kernels of PyTorch Models

24 May 2022

Papers citing "TorchNTK: A Library for Calculation of Neural Tangent Kernels of PyTorch Models"

3 / 3 papers shown

Title
Train Faster, Perform Better: Modular Adaptive Training in Over-Parameterized Models Yubin Shi Yixuan Chen Mingzhi Dong Xiaochen Yang Dongsheng Li ... Yingying Zhao Fan Yang Tun Lu Ning Gu L. Shang MoMe 46 4 0 13 May 2024
Characterizing the Spectrum of the NTK via a Power Series Expansion Michael Murray Hui Jin Benjamin Bowman Guido Montúfar 38 11 0 15 Nov 2022
Understanding Layer-wise Contributions in Deep Neural Networks through Spectral Analysis Yatin Dandi Arthur Jacot FAtt 23 4 0 06 Nov 2021