activity
20202024
most citedTorchNTK: A Library for Calculation of Neural Tangent Kernels of PyTorch Models

1 citations · 1 across the 2 of their papers we have counts for

collaborators
Showing cs.LGShow all

5 papers · 1 filter

cs.LG2024

Understanding Generative AI Content with Embedding Models

Max Vargas, Reilly Cannon, Andrew Engel +2

Constructing high-quality features is critical to any quantitative data analysis. While feature engineering was historically addressed by carefully hand-crafting data representatio…

cs.LG2023

Efficient kernel surrogates for neural network-based regression

Saad Qadeer, Andrew Engel, Amanda Howard +4

Despite their immense promise in performing a variety of learning tasks, a theoretical understanding of the limitations of Deep Neural Networks (DNNs) has so far eluded practitione…

cs.LG2023

Foundation Model's Embedded Representations May Detect Distribution Shift

Max Vargas, Adam Tsou, Andrew Engel +1

Sampling biases can cause distribution shifts between train and test datasets for supervised learning tasks, obscuring our ability to understand the generalization capacity of a mo…

cs.LG2023

Exploring Learned Representations of Neural Networks with Principal Component Analysis

Amit Harlev, Andrew Engel, Panos Stinis +1

Understanding feature representation for deep neural networks (DNNs) remains an open question within the general field of explainable AI. We use principal component analysis (PCA)…

cs.LG20221 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…