Zyphra / tree_attention

Related projects ⓘ

Alternatives and complementary repositories for tree_attention

• schwartz-lab-NLP / TOVA
  Token Omission Via Attention
  ☆120Updated last month
• HazyResearch / zoology
  Understand and test language model architectures on synthetic tasks.
  ☆162Updated 6 months ago
• athms / mad-lab
  A MAD laboratory to improve AI architecture designs 🧪
  ☆95Updated 6 months ago
• HazyResearch / based
  Code for exploring Based models from "Simple linear attention language models balance the recall-throughput tradeoff"
  ☆214Updated 3 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…
  ☆139Updated this week
• HazyResearch / lolcats
  Repo for "LoLCATs: On Low-Rank Linearizing of Large Language Models"
  ☆177Updated last month
• EleutherAI / nanoGPT-mup
  The simplest, fastest repository for training/finetuning medium-sized GPTs.
  ☆84Updated last week
• siyan-zhao / prepacking
  The source code of our work "Prepacking: A Simple Method for Fast Prefilling and Increased Throughput in Large Language Models"
  ☆56Updated last month
• llm-random / llm-random
  ☆175Updated this week
• jzhang38 / LongMamba
  Some preliminary explorations of Mamba's context scaling.
  ☆191Updated 9 months ago
• AnswerDotAI / cold-compress
  Cold Compress is a hackable, lightweight, and open-source toolkit for creating and benchmarking cache compression methods built on top of…
  ☆87Updated 3 months ago
• Zyphra / Zamba2
  PyTorch implementation of models from the Zamba2 series.
  ☆158Updated this week
• RobertCsordas / moeut
  ☆62Updated 3 months ago
• FasterDecoding / BitDelta
  ☆184Updated last month
• Infini-AI-Lab / MagicDec
  Breaking Throughput-Latency Trade-off for Long Sequences with Speculative Decoding
  ☆78Updated this week
• FasterDecoding / TEAL
  ☆96Updated last month
• cloneofsimo / min-max-gpt
  Minimal (400 LOC) implementation Maximum (multi-node, FSDP) GPT training
  ☆113Updated 7 months ago
• cloneofsimo / min-fsdp
  ☆73Updated 4 months ago
• kernelmachine / cbtm
  Code repository for the c-BTM paper
  ☆105Updated last year
• IST-DASLab / qmoe
  Code for the paper "QMoE: Practical Sub-1-Bit Compression of Trillion-Parameter Models".
  ☆262Updated last year
• jxiw / MambaInLlama
  [NeurIPS 2024] Official Repository of The Mamba in the Llama: Distilling and Accelerating Hybrid Models
  ☆174Updated this week
• shawntan / scattermoe
  Triton-based implementation of Sparse Mixture of Experts.
  ☆185Updated last month
• JacobPfau / fillerTokens
  ☆49Updated 6 months ago
• epfml / dynamic-sparse-flash-attention
  ☆132Updated last year
• tanaymeh / mamba-train
  A single repo with all scripts and utils to train / fine-tune the Mamba model with or without FIM
  ☆50Updated 7 months ago
• ScalingIntelligence / Archon
  Archon provides a modular framework for combining different inference-time techniques and LMs with just a JSON config file.
  ☆128Updated 3 weeks ago
• dvruette / barrel-rec-pytorch
  ☆53Updated 10 months ago
• ScalingIntelligence / large_language_monkeys
  ☆54Updated last month
• jxmorris12 / cde
  code for training & evaluating Contextual Document Embedding models
  ☆117Updated this week
• lucidrains / PEER-pytorch
  Pytorch implementation of the PEER block from the paper, Mixture of A Million Experts, by Xu Owen He at Deepmind
  ☆112Updated 2 months ago