Alternatives and similar repositories for flash-attention

Users that are interested in flash-attention are comparing it to the libraries listed below

• huggingface / peft
  🤗 PEFT: State-of-the-art Parameter-Efficient Fine-Tuning.
  ☆20,424Updated this week
• facebookresearch / xformers
  Hackable and optimized Transformers building blocks, supporting a composable construction.
  ☆10,257Updated this week
• huggingface / trl
  Train transformer language models with reinforcement learning.
  ☆16,915Updated this week
• bitsandbytes-foundation / bitsandbytes
  Accessible large language models via k-bit quantization for PyTorch.
  ☆7,881Updated this week
• huggingface / accelerate
  🚀 A simple way to launch, train, and use PyTorch models on almost any device and distributed configuration, automatic mixed precision (i…
  ☆9,427Updated this week
• NVIDIA / Megatron-LM
  Ongoing research training transformer models at scale
  ☆14,879Updated this week
• NVIDIA / FasterTransformer
  Transformer related optimization, including BERT, GPT
  ☆6,378Updated last year
• microsoft / LoRA
  Code for loralib, an implementation of "LoRA: Low-Rank Adaptation of Large Language Models"
  ☆13,142Updated last year
• huggingface / text-generation-inference
  Large Language Model Text Generation Inference
  ☆10,728Updated this week
• triton-lang / triton
  Development repository for the Triton language and compiler
  ☆18,098Updated this week
• EleutherAI / lm-evaluation-harness
  A framework for few-shot evaluation of language models.
  ☆11,123Updated this week
• BlinkDL / RWKV-LM
  RWKV (pronounced RwaKuv) is an RNN with great LLM performance, which can also be directly trained like a GPT transformer (parallelizable)…
  ☆14,273Updated 3 weeks ago
• sgl-project / sglang
  SGLang is a high-performance serving framework for large language models and multimodal models.
  ☆22,190Updated last week
• vllm-project / vllm
  A high-throughput and memory-efficient inference and serving engine for LLMs
  ☆67,159Updated this week
• artidoro / qlora
  QLoRA: Efficient Finetuning of Quantized LLMs
  ☆10,807Updated last year
• ggml-org / ggml
  Tensor library for machine learning
  ☆13,812Updated this week
• deepspeedai / DeepSpeed
  DeepSpeed is a deep learning optimization library that makes distributed training and inference easy, efficient, and effective.
  ☆41,208Updated this week
• NVIDIA / TensorRT-LLM
  TensorRT LLM provides users with an easy-to-use Python API to define Large Language Models (LLMs) and supports state-of-the-art optimizat…
  ☆12,588Updated this week
• microsoft / unilm
  Large-scale Self-supervised Pre-training Across Tasks, Languages, and Modalities
  ☆21,932Updated 3 weeks ago
• jzhang38 / TinyLlama
  The TinyLlama project is an open endeavor to pretrain a 1.1B Llama model on 3 trillion tokens.
  ☆8,867Updated last year
• OpenRLHF / OpenRLHF
  An Easy-to-use, Scalable and High-performance Agentic RL Framework based on Ray (PPO & DAPO & REINFORCE++ & TIS & vLLM & Ray & Async RL)
  ☆8,750Updated this week
• deepspeedai / DeepSpeedExamples
  Example models using DeepSpeed
  ☆6,761Updated 3 weeks ago
• InternLM / lmdeploy
  LMDeploy is a toolkit for compressing, deploying, and serving LLMs.
  ☆7,491Updated this week
• arcee-ai / mergekit
  Tools for merging pretrained large language models.
  ☆6,673Updated last week
• tatsu-lab / stanford_alpaca
  Code and documentation to train Stanford's Alpaca models, and generate the data.
  ☆30,269Updated last year
• mit-han-lab / llm-awq
  [MLSys 2024 Best Paper Award] AWQ: Activation-aware Weight Quantization for LLM Compression and Acceleration
  ☆3,410Updated 5 months ago
• mlfoundations / open_clip
  An open source implementation of CLIP.
  ☆13,223Updated 2 months ago
• meta-pytorch / torchtune
  PyTorch native post-training library
  ☆5,646Updated last week
• AutoGPTQ / AutoGPTQ
  An easy-to-use LLMs quantization package with user-friendly apis, based on GPTQ algorithm.
  ☆5,017Updated 9 months ago
• meta-pytorch / gpt-fast
  Simple and efficient pytorch-native transformer text generation in <1000 LOC of python.
  ☆6,175Updated 4 months ago