Alternatives and similar repositories for grokking

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

• ejmichaud / grokking-squared
  ☆25Updated 2 years ago
• thegregyang / NTK4A
  Code for the paper: "Tensor Programs II: Neural Tangent Kernel for Any Architecture"
  ☆106Updated 4 years ago
• Sea-Snell / grokking
  unofficial re-implementation of "Grokking: Generalization Beyond Overfitting on Small Algorithmic Datasets"
  ☆78Updated 2 years ago
• google / spectral-density
  Hessian spectral density estimation in TF and Jax
  ☆123Updated 4 years ago
• edwardjhu / TP4
  Code accompanying our paper "Feature Learning in Infinite-Width Neural Networks" (https://arxiv.org/abs/2011.14522)
  ☆62Updated 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
• ganguli-lab / degrees-of-freedom
  ☆37Updated 3 years ago
• aks2203 / deep-thinking
  A centralized place for deep thinking code and experiments
  ☆83Updated last year
• fwcore / mean-field-theory-deep-learning
  paper lists and information on mean-field theory of deep learning
  ☆74Updated 6 years ago
• js-d / sim_metric
  ☆35Updated last year
• akbir / deq-jax
  [NeurIPS'19] Deep Equilibrium Models Jax Implementation
  ☆39Updated 4 years ago
• stanislavfort / dissect-git-re-basin
  Replicating and dissecting the git-re-basin project in one-click-replication Colabs
  ☆36Updated 2 years ago
• KindXiaoming / Omnigrok
  Omnigrok: Grokking Beyond Algorithmic Data
  ☆56Updated 2 years ago
• bobby-he / Neural_Tangent_Kernel
  ☆66Updated 6 years ago
• 4rtemi5 / imax
  Image augmentation library for Jax
  ☆39Updated last year
• fKunstner / limitations-empirical-fisher
  Limitations of the Empirical Fisher Approximation
  ☆47Updated 2 months ago
• ethancaballero / broken_neural_scaling_laws
  Code Release for "Broken Neural Scaling Laws" (BNSL) paper
  ☆58Updated last year
• goldblum / TruthOrBackpropaganda
  An empirical investigation of deep learning theory
  ☆16Updated 5 years ago
• n2cholas / jax-resnet
  Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).
  ☆109Updated 2 years ago
• g-benton / loss-surface-simplexes
  ☆99Updated 3 years ago
• rahul13ramesh / compositional_capabilities
  Compositional Capabilities of Autoregressive Transformers: A Study on Synthetic, Interpretable Tasks
  ☆10Updated 10 months ago
• ppope / dimensions
  Code for "The Intrinsic Dimension of Images and Its Impact on Learning" - ICLR 2021 Spotlight https://openreview.net/forum?id=XJk19XzGq2J
  ☆68Updated last year
• thegregyang / NNspectra
  Computing the eigenvalues of Neural Tangent Kernel and Conjugate Kernel (aka NNGP kernel) over the boolean cube
  ☆47Updated 5 years ago
• locuslab / monotone_op_net
  Monotone operator equilibrium networks
  ☆52Updated 4 years ago
• nestordemeure / jochastic
  A JAX implementation of stochastic addition.
  ☆14Updated 2 years ago
• HazyResearch / structured-nets
  Structured matrices for compressing neural networks
  ☆66Updated last year
• KarenUllrich / binary-VAE
  A minimal implementation of a VAE with BinConcrete (relaxed Bernoulli) latent distribution in TensorFlow.
  ☆22Updated 5 years ago
• ganguli-lab / deepchaos
  Experiments for the paper "Exponential expressivity in deep neural networks through transient chaos"
  ☆71Updated 8 years ago
• CalculatedContent / ImplicitSelfRegularization
  Repo to accompany paper "Implicit Self-Regularization in Deep Neural Networks..."
  ☆44Updated 6 years ago
• cybertronai / autograd-lib
  ☆33Updated 4 years ago