Zyphra / tree_attention

Related projects: ⓘ

• HazyResearch / zoology
  Understand and test language model architectures on synthetic tasks.
  ☆156Updated 4 months ago
• schwartz-lab-NLP / TOVA
  Token Omission Via Attention
  ☆118Updated 7 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…
  ☆127Updated 2 weeks ago
• athms / mad-lab
  A MAD laboratory to improve AI architecture designs 🧪
  ☆84Updated 4 months ago
• HazyResearch / based
  Code for exploring Based models from "Simple linear attention language models balance the recall-throughput tradeoff"
  ☆206Updated last month
• AnswerDotAI / cold-compress
  Cold Compress is a hackable, lightweight, and open-source toolkit for creating and benchmarking cache compression methods built on top of…
  ☆73Updated last month
• cloneofsimo / min-max-gpt
  Minimal (400 LOC) implementation Maximum (multi-node, FSDP) GPT training
  ☆110Updated 5 months ago
• RobertCsordas / moeut
  ☆50Updated last month
• dvruette / barrel-rec-pytorch
  ☆53Updated 8 months ago
• siyan-zhao / prepacking
  The source code of our work "Prepacking: A Simple Method for Fast Prefilling and Increased Throughput in Large Language Models"
  ☆57Updated 5 months ago
• cloneofsimo / min-fsdp
  ☆68Updated 2 months ago
• llm-random / llm-random
  ☆169Updated this week
• mgmalek / efficient_cross_entropy
  ☆66Updated 3 months ago
• mcleish7 / arithmetic
  Code to reproduce "Transformers Can Do Arithmetic with the Right Embeddings", McLeish et al (2024)
  ☆169Updated 3 months ago
• MatX-inc / seqax
  seqax = sequence modeling + JAX
  ☆129Updated 2 months ago
• mobiusml / gemlite
  Simple and fast low-bit matmul kernels in CUDA
  ☆48Updated this week
• jxiw / MambaInLlama
  Official Repository of The Mamba in the Llama: Distilling and Accelerating Hybrid Models
  ☆130Updated this week
• berlino / seq_icl
  ☆48Updated 3 months ago
• kernelmachine / cbtm
  Code repository for the c-BTM paper
  ☆105Updated 11 months ago
• Zyphra / Zamba2
  PyTorch implementation of models from the Zamba2 series.
  ☆63Updated last month
• epfml / dynamic-sparse-flash-attention
  ☆129Updated last year
• lucidrains / PEER-pytorch
  Pytorch implementation of the PEER block from the paper, Mixture of A Million Experts, by Xu Owen He at Deepmind
  ☆105Updated 3 weeks ago
• jxbz / modula
  Scalable neural net training via automatic normalization in the modular norm.
  ☆108Updated 3 weeks ago
• shawntan / scattermoe
  Triton-based implementation of Sparse Mixture of Experts.
  ☆166Updated 3 weeks ago
• jzhang38 / LongMamba
  Some preliminary explorations of Mamba's context scaling.
  ☆184Updated 7 months ago
• mlfoundations / scaling
  Language models scale reliably with over-training and on downstream tasks
  ☆91Updated 5 months ago
• FasterDecoding / TEAL
  ☆75Updated this week
• minyoungg / LTE
  ☆61Updated 2 months ago
• EleutherAI / improved-t5
  Experiments for efforts to train a new and improved t5
  ☆76Updated 5 months ago
• foundation-model-stack / fms-fsdp
  🚀 Efficiently (pre)training foundation models with native PyTorch features, including FSDP for training and SDPA implementation of Flash…
  ☆150Updated last week