Alternatives and similar repositories for COSMOS

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

• RobertCsordas / moeut
  ☆91Updated last year
• berlino / seq_icl
  ☆53Updated last year
• JacobPfau / fillerTokens
  ☆74Updated last year
• zaydzuhri / softpick-attention
  Landing repository for the paper "Softpick: No Attention Sink, No Massive Activations with Rectified Softmax"
  ☆86Updated 4 months ago
• Edward-Sun / gpt-accelera
  Simple and efficient pytorch-native transformer training and inference (batched)
  ☆79Updated last year
• JoshEngels / MultiDimensionalFeatures
  Code for reproducing our paper "Not All Language Model Features Are Linear"
  ☆83Updated last year
• EleutherAI / nanoGPT-mup
  The simplest, fastest repository for training/finetuning medium-sized GPTs.
  ☆186Updated 3 weeks ago
• ScalingIntelligence / large_language_monkeys
  ☆112Updated last year
• schwartz-lab-NLP / TOVA
  Token Omission Via Attention
  ☆128Updated last year
• wdlctc / mini-s
  ☆54Updated last year
• kazuki-irie / kv-memory-brain
  Official Code Repository for the paper "Key-value memory in the brain"
  ☆31Updated 11 months ago
• UbiquantAI / URM
  Universal Reasoning Model
  ☆122Updated 3 weeks ago
• HazyResearch / train-tk
  train with kittens!
  ☆63Updated last year
• JonasGeiping / linear_cross_entropy_loss
  A fusion of a linear layer and a cross entropy loss, written for pytorch in triton.
  ☆75Updated last year
• Zyphra / tree_attention
  Tree Attention: Topology-aware Decoding for Long-Context Attention on GPU clusters
  ☆131Updated last year
• siyan-zhao / prepacking
  The source code of our work "Prepacking: A Simple Method for Fast Prefilling and Increased Throughput in Large Language Models" [AISTATS …
  ☆60Updated last year
• athms / mad-lab
  A MAD laboratory to improve AI architecture designs 🧪
  ☆137Updated last year
• OpenEvaByte / evabyte
  EvaByte: Efficient Byte-level Language Models at Scale
  ☆115Updated 9 months ago
• whyNLP / LCKV
  Layer-Condensed KV cache w/ 10 times larger batch size, fewer params and less computation. Dramatic speed up with better task performance…
  ☆157Updated 10 months ago
• JackCai1206 / arithmetic-self-improve
  ☆38Updated 11 months ago
• VITA-Group / WeLore
  [ICML 2025] From Low Rank Gradient Subspace Stabilization to Low-Rank Weights: Observations, Theories and Applications
  ☆52Updated 3 months ago
• gregorbachmann / Next-Token-Failures
  ☆108Updated last year
• jopetty / word-problem
  Experiments on the impact of depth in transformers and SSMs.
  ☆40Updated 3 months ago
• Zyphra / Zyda_processing
  ☆41Updated last year
• mnoukhov / async_rlhf
  Code and Configs for Asynchronous RLHF: Faster and More Efficient RL for Language Models
  ☆68Updated 9 months ago
• recursal / GoldFinch-paper
  GoldFinch and other hybrid transformer components
  ☆45Updated last year
• emalach / LinearLM
  Code for the paper: https://arxiv.org/pdf/2309.06979.pdf
  ☆21Updated last year
• epfml / schedules-and-scaling
  Code for NeurIPS 2024 Spotlight: "Scaling Laws and Compute-Optimal Training Beyond Fixed Training Durations"
  ☆88Updated last year
• LucasPrietoAl / grokking-at-the-edge-of-numerical-stability
  ☆111Updated 6 months ago
• facebookresearch / MemoryMosaics
  Memory Mosaics are networks of associative memories working in concert to achieve a prediction task.
  ☆57Updated last year