Dao-AILab / flash-attention
Fast and memory-efficient exact attention
β15,064Updated this week
Alternatives and similar repositories for flash-attention:
Users that are interested in flash-attention are comparing it to the libraries listed below
- Accessible large language models via k-bit quantization for PyTorch.β6,522Updated this week
- Hackable and optimized Transformers building blocks, supporting a composable construction.β8,910Updated this week
- π A simple way to launch, train, and use PyTorch models on almost any device and distributed configuration, automatic mixed precision (iβ¦β8,178Updated this week
- π€ PEFT: State-of-the-art Parameter-Efficient Fine-Tuning.β16,978Updated this week
- Transformer related optimization, including BERT, GPTβ5,981Updated 9 months ago
- Ongoing research training transformer models at scaleβ11,109Updated this week
- Code for loralib, an implementation of "LoRA: Low-Rank Adaptation of Large Language Models"β11,119Updated last month
- Train transformer language models with reinforcement learning.β10,609Updated this week
- QLoRA: Efficient Finetuning of Quantized LLMsβ10,168Updated 7 months ago
- A high-throughput and memory-efficient inference and serving engine for LLMsβ33,809Updated this week
- RWKV (pronounced RwaKuv) is an RNN with great LLM performance, which can also be directly trained like a GPT transformer (parallelizable)β¦β13,005Updated last week
- TensorRT-LLM provides users with an easy-to-use Python API to define Large Language Models (LLMs) and build TensorRT engines that containβ¦β9,147Updated this week
- SGLang is a fast serving framework for large language models and vision language models.β7,353Updated this week
- Large-scale Self-supervised Pre-training Across Tasks, Languages, and Modalitiesβ20,584Updated last week
- An open source implementation of CLIP.β10,804Updated last week
- Mamba SSM architectureβ13,791Updated last week
- A framework for few-shot evaluation of language models.β7,474Updated this week
- Large Language Model Text Generation Inferenceβ9,592Updated this week
- LAVIS - A One-stop Library for Language-Vision Intelligenceβ10,161Updated last month
- The TinyLlama project is an open endeavor to pretrain a 1.1B Llama model on 3 trillion tokens.β8,113Updated 8 months ago
- Implementation of the LLaMA language model based on nanoGPT. Supports flash attention, Int8 and GPTQ 4bit quantization, LoRA and LLaMA-Adβ¦β6,021Updated 4 months ago
- Development repository for the Triton language and compilerβ14,042Updated this week
- DeepSpeed is a deep learning optimization library that makes distributed training and inference easy, efficient, and effective.β36,255Updated this week
- Latest Advances on Multimodal Large Language Modelsβ13,556Updated this week
- [NeurIPS'23 Oral] Visual Instruction Tuning (LLaVA) built towards GPT-4V level capabilities and beyond.β21,096Updated 5 months ago
- PyTorch native post-training libraryβ4,703Updated this week
- Tensor library for machine learningβ11,541Updated this week
- Foundation Architecture for (M)LLMsβ3,038Updated 9 months ago
- [ICLR 2024] Efficient Streaming Language Models with Attention Sinksβ6,749Updated 6 months ago
- Inference Llama 2 in one file of pure Cβ17,858Updated 5 months ago