1 citations · 1 across the 1 of their papers we have counts for
1 paper
Andrew Engel, Zhichao Wang, Anand D. Sarwate +2
We introduce torchNTK, a python library to calculate the empirical neural tangent kernel (NTK) of neural network models in the PyTorch framework. We provide an efficient method to…