neural-tangent-kernel topic

List neural-tangent-kernel repositories

nngeometry

196
Stars
20
Forks
Watchers

{KFAC,EKFAC,Diagonal,Implicit} Fisher Matrices and finite width NTKs in PyTorch

eigenlearning

49
Stars
8
Forks
Watchers

codebase for "A Theory of the Inductive Bias and Generalization of Kernel Regression and Wide Neural Networks"

awesome-deep-phenomena

214
Stars
8
Forks
Watchers

A curated list of papers of interesting empirical study and insight on deep learning. Continually updating...

dks

52
Stars
2
Forks
Watchers

Multi-framework implementation of Deep Kernel Shaping and Tailored Activation Transformations, which are methods that modify neural network models (and their initializations) to make them easier to tr...

Lossless_Compression

21
Stars
2
Forks
Watchers

We propose a lossless compression algorithm based on the NTK matrix for DNN. The compressed network yields asymptotically the same NTK as the original (dense and unquantized) network, with its weight...

scikit-ntk

21
Stars
0
Forks
Watchers

Neural Tangent Kernel (NTK) module for the scikit-learn library

Neural-Thompson-Sampling

18
Stars
5
Forks
Watchers

Study of the paper 'Neural Thompson Sampling' published in October 2020

trak

141
Stars
18
Forks
Watchers

A fast, effective data attribution method for neural networks in PyTorch