Alternatives and similar repositories for grokking

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

• edwardjhu / TP4
  Code accompanying our paper "Feature Learning in Infinite-Width Neural Networks" (https://arxiv.org/abs/2011.14522)
  ☆62Updated 4 years ago
• Sea-Snell / grokking
  unofficial re-implementation of "Grokking: Generalization Beyond Overfitting on Small Algorithmic Datasets"
  ☆78Updated 2 years ago
• thegregyang / NTK4A
  Code for the paper: "Tensor Programs II: Neural Tangent Kernel for Any Architecture"
  ☆105Updated 4 years ago
• ejmichaud / grokking-squared
  ☆26Updated 2 years ago
• aradha / recursive_feature_machines
  ☆51Updated last month
• google / spectral-density
  Hessian spectral density estimation in TF and Jax
  ☆123Updated 4 years ago
• thegregyang / NNspectra
  Computing the eigenvalues of Neural Tangent Kernel and Conjugate Kernel (aka NNGP kernel) over the boolean cube
  ☆47Updated 5 years ago
• akbir / deq-jax
  [NeurIPS'19] Deep Equilibrium Models Jax Implementation
  ☆39Updated 4 years ago
• aks2203 / easy-to-hard
  Official repository for the paper "Can You Learn an Algorithm? Generalizing from Easy to Hard Problems with Recurrent Networks"
  ☆59Updated 3 years ago
• lixilinx / psgd_torch
  Pytorch implementation of preconditioned stochastic gradient descent (Kron and affine preconditioner, low-rank approximation precondition…
  ☆175Updated this week
• g-benton / hessian-eff-dim
  Public Codebase for Rethinking Parameter Counting: Effective Dimensionality Revisited
  ☆37Updated 2 years ago
• KindXiaoming / Omnigrok
  Omnigrok: Grokking Beyond Algorithmic Data
  ☆58Updated 2 years ago
• KarenUllrich / binary-VAE
  A minimal implementation of a VAE with BinConcrete (relaxed Bernoulli) latent distribution in TensorFlow.
  ☆22Updated 5 years ago
• ExpectationMax / sklearn-jax-kernels
  Composable kernels for scikit-learn implemented in JAX.
  ☆43Updated 4 years ago
• g-benton / loss-surface-simplexes
  ☆99Updated 3 years ago
• ganguli-lab / degrees-of-freedom
  ☆37Updated 3 years ago
• Ping-C / optimizer
  This repository includes code to reproduce the tables in "Loss Landscapes are All You Need: Neural Network Generalization Can Be Explaine…
  ☆37Updated 2 years ago
• tk-rusch / unicornn
  Official code for UnICORNN (ICML 2021)
  ☆27Updated 3 years ago
• gkdziugaite / pacbayes-opt
  Optimizing PAC-Bayes bounds for Stochastic Neural Networks with Gaussian weights
  ☆27Updated 4 years ago
• gehring / fax
  ☆80Updated 3 years ago
• CalculatedContent / ImplicitSelfRegularization
  Repo to accompany paper "Implicit Self-Regularization in Deep Neural Networks..."
  ☆44Updated 6 years ago
• gebob19 / introduction_to_normalizing_flows
  Jupyter Notebook corresponding to 'Going with the Flow: An Introduction to Normalizing Flows'
  ☆26Updated 4 years ago
• fwcore / mean-field-theory-deep-learning
  paper lists and information on mean-field theory of deep learning
  ☆76Updated 6 years ago
• HazyResearch / structured-nets
  Structured matrices for compressing neural networks
  ☆66Updated last year
• james-simon / eigenlearning
  codebase for "A Theory of the Inductive Bias and Generalization of Kernel Regression and Wide Neural Networks"
  ☆49Updated 2 years ago
• cybertronai / autograd-lib
  ☆33Updated 5 years ago
• RobertTLange / code-and-blog
  Some small scale experiments for my blog posts 📝
  ☆79Updated 2 years ago
• mansheej / lth_diet
  ☆10Updated 2 years ago
• facebookresearch / semi-discrete-flow
  code for "Semi-Discrete Normalizing Flows through Differentiable Tessellation"
  ☆26Updated 2 years ago
• dylandoblar / noether-networks
  Meta-learning inductive biases in the form of useful conserved quantities.
  ☆37Updated 2 years ago