Alternatives and similar repositories for Omnigrok:

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

• danielmamay / grokking
  Implementation of OpenAI's 'Grokking: Generalization Beyond Overfitting on Small Algorithmic Datasets' paper.
  ☆35Updated last year
• Sea-Snell / grokking
  unofficial re-implementation of "Grokking: Generalization Beyond Overfitting on Small Algorithmic Datasets"
  ☆72Updated 2 years ago
• locuslab / edge-of-stability
  ☆65Updated 2 months ago
• ejmichaud / grokking-squared
  ☆24Updated 2 years ago
• wiseodd / pub-ready-plots
  Agustinus' very opiniated publication-ready plotting library
  ☆61Updated last month
• jongharyu / neural-svd
  ☆16Updated 5 months ago
• dtsip / in-context-learning
  ☆213Updated 9 months ago
• KindXiaoming / BIMT
  Brain-Inspired Modular Training (BIMT), a method for making neural networks more modular and interpretable.
  ☆165Updated last year
• Ping-C / optimizer
  This repository includes code to reproduce the tables in "Loss Landscapes are All You Need: Neural Network Generalization Can Be Explaine…
  ☆35Updated 2 years ago
• thegregyang / NTK4A
  Code for the paper: "Tensor Programs II: Neural Tangent Kernel for Any Architecture"
  ☆105Updated 4 years ago
• xu-ji / information-bottleneck
  Deep Learning & Information Bottleneck
  ☆57Updated last year
• formll / dog
  DoG is SGD's Best Friend: A Parameter-Free Dynamic Step Size Schedule
  ☆59Updated last year
• lixilinx / psgd_torch
  Pytorch implementation of preconditioned stochastic gradient descent (Kron and affine preconditioner, low-rank approximation precondition…
  ☆168Updated 2 months ago
• lee-ny / teaching_arithmetic
  ☆78Updated last year
• ApolloResearch / e2e_sae
  Sparse Autoencoder Training Library
  ☆42Updated 4 months ago
• f-dangel / curvlinops
  PyTorch linear operators for curvature matrices (Hessian, Fisher/GGN, KFAC, ...)
  ☆32Updated this week
• teddykoker / grokking
  PyTorch implementation of "Grokking: Generalization Beyond Overfitting on Small Algorithmic Datasets"
  ☆35Updated 3 years ago
• jacobdunefsky / transcoder_circuits
  ☆57Updated 3 months ago
• michaelsdr / sinkformers
  Transformers with doubly stochastic attention
  ☆45Updated 2 years ago
• wesg52 / universal-neurons
  Universal Neurons in GPT2 Language Models
  ☆27Updated 9 months ago
• mechanistic-interpretability-grokking / progress-measures-paper
  ☆61Updated 2 years ago
• aks2203 / deep-thinking
  A centralized place for deep thinking code and experiments
  ☆82Updated last year
• redwoodresearch / Easy-Transformer
  ☆110Updated 6 months ago
• stanislavfort / dissect-git-re-basin
  Replicating and dissecting the git-re-basin project in one-click-replication Colabs
  ☆36Updated 2 years ago
• necludov / jam
  Implementation of Action Matching
  ☆38Updated last year
• kwignb / NeuralTangentKernel-Papers
  Neural Tangent Kernel Papers
  ☆106Updated last month
• AllanYangZhou / universal_neural_functional
  ☆49Updated last year
• nikhilvyas / SOAP
  ☆161Updated 3 months ago
• wattenberg / superposition
  Code associated to papers on superposition (in ML interpretability)
  ☆28Updated 2 years ago
• pomonam / kronfluence
  Influence Functions with (Eigenvalue-corrected) Kronecker-Factored Approximate Curvature
  ☆136Updated 7 months ago