CLAIRE-Labo / flash_attention

Related projects ⓘ

Alternatives and complementary repositories for flash_attention

• cloneofsimo / min-fsdp
  ☆72Updated 4 months ago
• srush / mamba-primer
  ☆35Updated 7 months ago
• epfml / schedules-and-scaling
  ☆50Updated 2 weeks ago
• dvruette / barrel-rec-pytorch
  ☆53Updated 9 months ago
• cloneofsimo / min-max-gpt
  Minimal (400 LOC) implementation Maximum (multi-node, FSDP) GPT training
  ☆112Updated 6 months ago
• EleutherAI / nanoGPT-mup
  The simplest, fastest repository for training/finetuning medium-sized GPTs.
  ☆84Updated 2 weeks ago
• shreyansh26 / Attention-Mask-Patterns
  Using FlexAttention to compute attention with different masking patterns
  ☆40Updated last month
• epfml / DenseFormer
  ☆76Updated 6 months ago
• Edward-Sun / gpt-accelera
  Simple and efficient pytorch-native transformer training and inference (batched)
  ☆61Updated 7 months ago
• young-geng / mintext
  Minimal but scalable implementation of large language models in JAX
  ☆25Updated last week
• amirzandieh / HyperAttention
  Triton Implementation of HyperAttention Algorithm
  ☆46Updated 11 months ago
• dangxingyu / rnn-icrag
  Official repository of paper "RNNs Are Not Transformers (Yet): The Key Bottleneck on In-context Retrieval"
  ☆24Updated 6 months ago
• mlfoundations / scaling
  Language models scale reliably with over-training and on downstream tasks
  ☆94Updated 7 months ago
• joey00072 / ohara
  Collection of autoregressive model implementation
  ☆66Updated last week
• athms / mad-lab
  A MAD laboratory to improve AI architecture designs 🧪
  ☆95Updated 6 months ago
• RobertCsordas / moeut
  ☆61Updated 2 months ago
• ethansmith2000 / TransformerExperiments
  ☆18Updated last month
• RobertCsordas / moe
  Official repository for the paper "Approximating Two-Layer Feedforward Networks for Efficient Transformers"
  ☆36Updated 11 months ago
• ScalingIntelligence / large_language_monkeys
  ☆50Updated last month
• recursal / GoldFinch-paper
  GoldFinch and other hybrid transformer components
  ☆39Updated 3 months ago
• berlino / seq_icl
  ☆50Updated 5 months ago
• sjelassi / transformers_ssm_copy
  ☆24Updated 8 months ago
• jiasenlu / LL3M
  LL3M: Large Language and Multi-Modal Model in Jax
  ☆64Updated 6 months ago
• cloneofsimo / zeroshampoo
  ☆31Updated 2 months ago
• google-deepmind / spectral_ssm
  ☆27Updated 7 months ago
• srush / triton-autodiff
  Experiment of using Tangent to autodiff triton
  ☆72Updated 9 months ago
• ml-jku / EVA
  One Initialization to Rule them All: Fine-tuning via Explained Variance Adaptation
  ☆29Updated 3 weeks ago
• idiap / sigma-gpt
  σ-GPT: A New Approach to Autoregressive Models
  ☆59Updated 2 months ago
• lucidrains / taylor-series-linear-attention
  Explorations into the recently proposed Taylor Series Linear Attention
  ☆89Updated 2 months ago