Alternatives and similar repositories for ntk

Users that are interested in ntk are comparing it to the libraries listed below

• ruosongwang / CNTK
  Convolutional Neural Tangent Kernel
  ☆112Updated 6 years ago
• bobby-he / Neural_Tangent_Kernel
  ☆67Updated 6 years ago
• fwcore / mean-field-theory-deep-learning
  paper lists and information on mean-field theory of deep learning
  ☆78Updated 6 years ago
• Thrandis / EKFAC-pytorch
  Repository containing Pytorch code for EKFAC and K-FAC perconditioners.
  ☆147Updated 2 years ago
• rajatvd / NTK
  Code for experiments in my blog post on the Neural Tangent Kernel: https://eigentales.com/NTK
  ☆176Updated 5 years ago
• noahgolmant / pytorch-hessian-eigenthings
  Efficient PyTorch Hessian eigendecomposition tools!
  ☆379Updated last year
• cybertronai / autograd-hacks
  ☆157Updated 3 years ago
• thegregyang / NTK4A
  Code for the paper: "Tensor Programs II: Neural Tangent Kernel for Any Architecture"
  ☆104Updated 5 years ago
• wiseodd / natural-gradients
  Collection of algorithms for approximating Fisher Information Matrix for Natural Gradient (and second order method in general)
  ☆141Updated 6 years ago
• timgaripov / dnn-mode-connectivity
  Mode Connectivity and Fast Geometric Ensembles in PyTorch
  ☆278Updated 3 years ago
• alecwangcq / KFAC-Pytorch
  Pytorch implementation of KFAC and E-KFAC (Natural Gradient).
  ☆131Updated 6 years ago
• tfjgeorge / nngeometry
  {KFAC,EKFAC,Diagonal,Implicit} Fisher Matrices and finite width NTKs in PyTorch
  ☆215Updated last month
• prolearner / hypertorch
  ☆124Updated last year
• gbaydin / hypergradient-descent
  Hypergradient descent
  ☆148Updated last year
• artemyk / ibsgd
  ☆154Updated 5 years ago
• yaodongyu / Rethink-BiasVariance-Tradeoff
  Rethinking Bias-Variance Trade-off for Generalization of Neural Networks
  ☆50Updated 4 years ago
• umutsimsekli / sgd_tail_index
  The codebase for the paper "A Tail-Index Analysis of Stochastic Gradient Noise in Deep Neural Networks"
  ☆25Updated 6 years ago
• LeoYu / neural-tangent-kernel-UCI
  Testing Nerual Tangent Kernel (NTK) on small UCI datasets
  ☆80Updated 6 years ago
• google / spectral-density
  Hessian spectral density estimation in TF and Jax
  ☆124Updated 5 years ago
• reiinakano / invariant-risk-minimization
  Implementation of Invariant Risk Minimization https://arxiv.org/abs/1907.02893
  ☆91Updated 5 years ago
• cybertronai / pytorch-sso
  PyTorch-SSO: Scalable Second-Order methods in PyTorch
  ☆147Updated 2 years ago
• ruqizhang / csgmcmc
  Cyclical Stochastic Gradient MCMC for Bayesian Deep Learning
  ☆92Updated 4 years ago
• tudor-berariu / fisher-information-matrix
  PyTorch implementation of FIM and empirical FIM
  ☆61Updated 7 years ago
• mariogeiger / hessian
  hessian in pytorch
  ☆187Updated 4 years ago
• gibipara92 / learning-explanations-hard-to-vary
  Code to implement the AND-mask and geometric mean to do gradient based optimization, from the paper "Learning explanations that are hard …
  ☆41Updated 4 years ago
• wronnyhuang / gen-viz
  Code for the paper "Understanding Generalization through Visualizations"
  ☆64Updated 4 years ago
• leiwu0 / sgd.stability
  Analyze the dynamic stability of SGD
  ☆12Updated 6 years ago
• fKunstner / limitations-empirical-fisher
  Limitations of the Empirical Fisher Approximation
  ☆48Updated 7 months ago
• ganguli-lab / deepchaos
  Experiments for the paper "Exponential expressivity in deep neural networks through transient chaos"
  ☆73Updated 9 years ago
• alexalemi / vib_demo
  ☆91Updated 3 years ago