google / neural-tangents Star 2.2k Code Issues Pull requests Discussions Fast and Easy Infinite Neural Networks in Python kernel neural-networks gradient-descent bayesian-inference gaussian-processes bayesian-networks deep-networks gradient-flow jax infinite-networks training-dynamics neural-tangents kernel-computation Updated Mar 1, 2024 Jupyter Notebook
james-simon / eigenlearning Star 49 Code Issues Pull requests codebase for "A Theory of the Inductive Bias and Generalization of Kernel Regression and Wide Neural Networks" deep-learning eigenvalues deep-learning-theory neural-tangents neural-tangent-kernel Updated May 2, 2023 Jupyter Notebook
AlbertDominguez / infinite-width-NNs Star 6 Code Issues Pull requests Infinite-width neural networks from a practical point of view kernel-methods ntk gaussian-process-regression mlp-networks jax infinite-networks neural-tangents Updated Feb 24, 2021 Python