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"
  ☆70Updated 2 years ago
• ejmichaud / grokking-squared
  ☆24Updated last year
• locuslab / edge-of-stability
  ☆63Updated last month
• mechanistic-interpretability-grokking / progress-measures-paper
  ☆59Updated 2 years ago
• xu-ji / information-bottleneck
  Deep Learning & Information Bottleneck
  ☆53Updated last year
• teddykoker / grokking
  PyTorch implementation of "Grokking: Generalization Beyond Overfitting on Small Algorithmic Datasets"
  ☆34Updated 3 years ago
• aw31 / empirical-ntks
  Efficient empirical NTKs in PyTorch
  ☆18Updated 2 years ago
• formll / dog
  DoG is SGD's Best Friend: A Parameter-Free Dynamic Step Size Schedule
  ☆58Updated last year
• bilal-chughtai / rep-theory-mech-interp
  ☆26Updated last year
• KihoPark / linear_rep_geometry
  ☆83Updated last year
• stanislavfort / dissect-git-re-basin
  Replicating and dissecting the git-re-basin project in one-click-replication Colabs
  ☆36Updated 2 years ago
• pomonam / kronfluence
  Influence Functions with (Eigenvalue-corrected) Kronecker-Factored Approximate Curvature
  ☆127Updated 5 months ago
• chengxiang / LinearTransformer
  Pytorch code for experiments on Linear Transformers
  ☆17Updated last year
• aks2203 / deep-thinking
  A centralized place for deep thinking code and experiments
  ☆79Updated last year
• dtsip / in-context-learning
  ☆211Updated 8 months ago
• adamkarvonen / SAEBench
  ☆45Updated this week
• michaelsdr / sinkformers
  Transformers with doubly stochastic attention
  ☆44Updated 2 years ago
• thegregyang / NTK4A
  Code for the paper: "Tensor Programs II: Neural Tangent Kernel for Any Architecture"
  ☆103Updated 4 years ago
• mansheej / icl-task-diversity
  Code for the paper "Pretraining task diversity and the emergence of non-Bayesian in-context learning for regression"
  ☆20Updated last year
• wesg52 / universal-neurons
  Universal Neurons in GPT2 Language Models
  ☆27Updated 8 months ago
• ApolloResearch / e2e_sae
  Sparse Autoencoder Training Library
  ☆39Updated 3 months ago
• autonomousvision / akorn
  Artificial Kuramoto Oscillatory Neurons
  ☆46Updated last week
• vvvm23 / mamba-jax
  Unofficial but Efficient Implementation of "Mamba: Linear-Time Sequence Modeling with Selective State Spaces" in JAX
  ☆82Updated last year
• aadityasingh / icl-dynamics
  ☆16Updated 9 months ago
• KindXiaoming / BIMT
  Brain-Inspired Modular Training (BIMT), a method for making neural networks more modular and interpretable.
  ☆164Updated last year
• Silent-Zebra / twisted-smc-lm
  ☆24Updated last week
• redwoodresearch / Easy-Transformer
  ☆109Updated 5 months ago
• UniReps / UniReps-resources
  ☆17Updated last year
• aradha / recursive_feature_machines
  ☆41Updated this week