Alternatives and similar repositories for structural-grokking

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

• abhishekpanigrahi1996 / transformer_in_transformer
  ☆45Updated 2 years ago
• gregorbachmann / Next-Token-Failures
  ☆101Updated last year
• MurtyShikhar / Pushdown-Layers
  Code for Pushdown Layers from our EMNLP 2023 paper
  ☆29Updated last year
• dangxingyu / rnn-icrag
  Official repository of paper "RNNs Are Not Transformers (Yet): The Key Bottleneck on In-context Retrieval"
  ☆27Updated last year
• Edward-Sun / gpt-accelera
  Simple and efficient pytorch-native transformer training and inference (batched)
  ☆78Updated last year
• epfml / schedules-and-scaling
  Code for NeurIPS 2024 Spotlight: "Scaling Laws and Compute-Optimal Training Beyond Fixed Training Durations"
  ☆84Updated 11 months ago
• berlino / seq_icl
  ☆53Updated last year
• jopetty / word-problem
  Experiments on the impact of depth in transformers and SSMs.
  ☆35Updated 11 months ago
• sjelassi / transformers_ssm_copy
  ☆33Updated last year
• sustcsonglin / mamba-triton
  ☆48Updated last year
• HKUNLP / diffusion-vs-ar
  [ICLR 2025] Code for the paper "Beyond Autoregression: Discrete Diffusion for Complex Reasoning and Planning"
  ☆77Updated 8 months ago
• IBM / ColPret
  Efficient Scaling laws and collaborative pretraining.
  ☆18Updated last month
• shikaiqiu / compute-better-spent
  ☆58Updated last year
• shawntan / stickbreaking-attention
  Stick-breaking attention
  ☆60Updated 3 months ago
• facebookresearch / MemoryMosaics
  Memory Mosaics are networks of associative memories working in concert to achieve a prediction task.
  ☆48Updated 8 months ago
• nouhadziri / faith-and-fate
  ☆37Updated last year
• google-deepmind / emergent_in_context_learning
  ☆85Updated last year
• renll / SeqBoat
  [NeurIPS 2023] Sparse Modular Activation for Efficient Sequence Modeling
  ☆39Updated last year
• shreyansh26 / Attention-Mask-Patterns
  Using FlexAttention to compute attention with different masking patterns
  ☆46Updated last year
• dvruette / barrel-rec-pytorch
  ☆53Updated last year
• janphilippfranken / sami
  Self-Supervised Alignment with Mutual Information
  ☆21Updated last year
• mlfoundations / scaling
  Language models scale reliably with over-training and on downstream tasks
  ☆100Updated last year
• jxiw / BiGS
  Official Repository of Pretraining Without Attention (BiGS), BiGS is the first model to achieve BERT-level transfer learning on the GLUE …
  ☆114Updated last year
• katiekang1998 / reasoning_generalization
  ☆33Updated 9 months ago
• microsoft / SparseMixer
  Sparse Backpropagation for Mixture-of-Expert Training
  ☆29Updated last year
• haozheji / exact-optimization
  ICML 2024 - Official Repository for EXO: Towards Efficient Exact Optimization of Language Model Alignment
  ☆57Updated last year
• nathanhu0 / CaMeLS
  Codebase for Context-aware Meta-learned Loss Scaling (CaMeLS). https://arxiv.org/abs/2305.15076.
  ☆25Updated last year
• AndPotap / einsum-search
  ☆33Updated last year
• srush / mamba-scans
  Blog post
  ☆17Updated last year
• ylsung / vl-merging
  PyTorch codes for the paper "An Empirical Study of Multimodal Model Merging"
  ☆37Updated 2 years ago