1 citations · 3 across the 10 of their papers we have counts for
Showing 2022Show all
2 papers · 1 filter
cs.LG2022
Spectral Evolution and Invariance in Linear-width Neural Networks
Zhichao Wang, Andrew Engel, Anand Sarwate +2
We investigate the spectral properties of linear-width feed-forward neural networks, where the sample size is asymptotically proportional to network width. Empirically, we show tha…
cs.LG2022★ 1 cited
TorchNTK: A Library for Calculation of Neural Tangent Kernels of PyTorch Models
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…