Alternatives and similar repositories for grokking-squared:

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

• ethancaballero / broken_neural_scaling_laws
  Code Release for "Broken Neural Scaling Laws" (BNSL) paper
  ☆58Updated last year
• dylandoblar / noether-networks
  Meta-learning inductive biases in the form of useful conserved quantities.
  ☆37Updated 2 years ago
• bhoov / energy-transformer-jax
  The Energy Transformer block, in JAX
  ☆56Updated last year
• bilal-chughtai / rep-theory-mech-interp
  ☆26Updated last year
• GFNOrg / EB_GFN
  Code for our paper "Generative Flow Networks for Discrete Probabilistic Modeling"
  ☆81Updated last year
• vvvm23 / mamba-jax
  Unofficial but Efficient Implementation of "Mamba: Linear-Time Sequence Modeling with Selective State Spaces" in JAX
  ☆82Updated last year
• AndPotap / einsum-search
  ☆29Updated 4 months ago
• choidami / sst
  ☆49Updated 4 years ago
• GFNOrg / GFlowNet-EM
  Code for GFlowNet-EM, a novel algorithm for fitting latent variable models with compositional latents and an intractable true posterior.
  ☆41Updated last year
• shikaiqiu / compute-better-spent
  ☆52Updated 4 months ago
• Sea-Snell / grokking
  unofficial re-implementation of "Grokking: Generalization Beyond Overfitting on Small Algorithmic Datasets"
  ☆72Updated 2 years ago
• yilundu / improved_contrastive_divergence
  [ICML'21] Improved Contrastive Divergence Training of Energy Based Models
  ☆62Updated 2 years ago
• RobertCsordas / modules
  The official repository for our paper "Are Neural Nets Modular? Inspecting Functional Modularity Through Differentiable Weight Masks". We…
  ☆46Updated last year
• tristandeleu / jax-dag-gflownet
  Code for "Bayesian Structure Learning with Generative Flow Networks"
  ☆83Updated 2 years ago
• augustwester / gflownet
  A PyTorch implementation of a Generative Flow Network (GFlowNet) proposed by Bengio et al. (2021)
  ☆42Updated last year
• ssokota / mec
  Code for minimum-entropy coupling.
  ☆31Updated 7 months ago
• akbir / deq-jax
  [NeurIPS'19] Deep Equilibrium Models Jax Implementation
  ☆39Updated 4 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…
  ☆35Updated last year
• 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
• danielmamay / grokking
  Implementation of OpenAI's 'Grokking: Generalization Beyond Overfitting on Small Algorithmic Datasets' paper.
  ☆35Updated last year
• IDSIA / recurrent-fwp
  Official repository for the paper "Going Beyond Linear Transformers with Recurrent Fast Weight Programmers" (NeurIPS 2021)
  ☆48Updated last year
• MichaelArbel / GeneralizedEBM
  ☆53Updated 6 months ago
• wgrathwohl / GWG_release
  Official release of code for "Oops I Took A Gradient: Scalable Sampling for Discrete Distributions"
  ☆54Updated last year
• tung-nd / TNP-pytorch
  Official implementation of Transformer Neural Processes
  ☆71Updated 2 years ago
• zdhNarsil / Distributional-GFlowNets
  Code for our TMLR paper "Distributional GFlowNets with Quantile Flows".
  ☆9Updated last year
• runame / laplace-redux
  Laplace Redux -- Effortless Bayesian Deep Learning
  ☆41Updated last year
• locuslab / monotone_op_net
  Monotone operator equilibrium networks
  ☆51Updated 4 years ago
• wouterkool / estimating-gradients-without-replacement
  Estimating Gradients for Discrete Random Variables by Sampling without Replacement
  ☆40Updated 5 years ago
• team-approx-bayes / fromp
  Contains code for the NeurIPS 2020 paper by Pan et al., "Continual Deep Learning by FunctionalRegularisation of Memorable Past"
  ☆44Updated 4 years ago