rajatvd / NTKView external linksLinks
Code for experiments in my blog post on the Neural Tangent Kernel: https://eigentales.com/NTK
☆173Nov 10, 2019Updated 6 years ago
Alternatives and similar repositories for NTK
Users that are interested in NTK are comparing it to the libraries listed below
Sorting:
- Convolutional Neural Tangent Kernel☆112Oct 22, 2019Updated 6 years ago
- Fast and Easy Infinite Neural Networks in Python☆2,374Mar 1, 2024Updated last year
- NTK reading group☆87Nov 14, 2019Updated 6 years ago
- Code for the paper: "Tensor Programs II: Neural Tangent Kernel for Any Architecture"☆109Aug 26, 2020Updated 5 years ago
- ☆68Mar 15, 2019Updated 6 years ago
- Layered distributions using FLAX/JAX☆10Dec 13, 2020Updated 5 years ago
- Code accompanying our paper "Finding trainable sparse networks through Neural Tangent Transfer" to be published at ICML-2020.☆13Jun 14, 2020Updated 5 years ago
- Testing Nerual Tangent Kernel (NTK) on small UCI datasets☆80Oct 7, 2019Updated 6 years ago
- Example code for the NNGeometry PyTorch library☆10Aug 20, 2025Updated 5 months ago
- ☆15Sep 19, 2019Updated 6 years ago
- paper lists and information on mean-field theory of deep learning☆79Mar 25, 2019Updated 6 years ago
- Implementation of Physics-Informed PointNet (PIPN) for weakly-supervised learning of 2D linear elasticity (plane stress) on multiple sets…☆12Apr 11, 2024Updated last year
- Code to accompany paper 'Bayesian Deep Ensembles via the Neural Tangent Kernel'☆26Mar 12, 2021Updated 4 years ago
- Refining continuous-in-depth neural networks☆42Nov 14, 2021Updated 4 years ago
- QCHS2022 Summer School Tutorial for Yao and Bloqade☆13Jan 13, 2023Updated 3 years ago
- Code for NeurIPS 2019 paper: "Tensor Programs I: Wide Feedforward or Recurrent Neural Networks of Any Architecture are Gaussian Processes…☆249Aug 26, 2020Updated 5 years ago
- Repo to accompany paper "Implicit Self-Regularization in Deep Neural Networks..."☆47Jan 25, 2019Updated 7 years ago
- Implementation of Graph Neural Tangent Kernel (NeurIPS 2019)☆105Jan 28, 2020Updated 6 years ago
- We simulate a wind tunnel, place a rectangular occlusion in it, and then use gradient descent to turn the occlusion into a wing.☆27Oct 16, 2020Updated 5 years ago
- Collection of algorithms for approximating Fisher Information Matrix for Natural Gradient (and second order method in general)☆142May 26, 2019Updated 6 years ago
- Code for "'Hey, that's not an ODE:' Faster ODE Adjoints via Seminorms" (ICML 2021)☆91Oct 19, 2022Updated 3 years ago
- DropNet: Reducing Neural Network Complexity via Iterative Pruning (ICML 2020)☆16Aug 24, 2020Updated 5 years ago
- Efficient PyTorch Hessian eigendecomposition tools!☆384Feb 29, 2024Updated last year
- PyTorch Code for "Evaluating the search phase of Neural Architecture Search" @ ICLR 2020☆49Dec 17, 2019Updated 6 years ago
- ☆68Oct 13, 2022Updated 3 years ago
- Episodic Control☆21Sep 20, 2022Updated 3 years ago
- Differentiable interface to FEniCS/Firedrake for JAX using dolfin-adjoint/pyadjoint☆107Nov 12, 2023Updated 2 years ago
- Computing the eigenvalues of Neural Tangent Kernel and Conjugate Kernel (aka NNGP kernel) over the boolean cube☆48Sep 19, 2019Updated 6 years ago
- ☆168Jun 27, 2022Updated 3 years ago
- Practical Machine Learning With R and Python☆10Dec 3, 2017Updated 8 years ago
- Pytorch implementation of Neural Processes for functions and images☆236Feb 8, 2022Updated 4 years ago
- Model Agnostic Meta Learning (MAML) implemented in Flax, the neural network library for JAX.☆21Sep 18, 2020Updated 5 years ago
- ☆32Aug 22, 2020Updated 5 years ago
- Deep neural network kernel for Gaussian process☆212Sep 8, 2020Updated 5 years ago
- Unsupervised machine learning of phase of matter in physics☆11Feb 12, 2017Updated 9 years ago
- Dataset and code to reproduce the results of the paper "Evolving Structures in Complex Systems"☆11Dec 16, 2019Updated 6 years ago
- B.Tech Project On demodulation technique of OTFS(Orthogonal Time Frequency Space) at imperfect Channel State Information and lower SNR(dB…☆11May 12, 2024Updated last year
- ☆11Apr 14, 2022Updated 3 years ago
- Compressed ML Math material based on the book "Mathematics for Machine Learning" and other resources.☆10Jan 7, 2020Updated 6 years ago