Dao-AILab / flash-attention
Fast and memory-efficient exact attention
☆13,401Updated this week
Related projects: ⓘ
- 🤗 PEFT: State-of-the-art Parameter-Efficient Fine-Tuning.☆15,839Updated this week
- 🚀 A simple way to launch, train, and use PyTorch models on almost any device and distributed configuration, automatic mixed precision (i…☆7,687Updated this week
- Code for loralib, an implementation of "LoRA: Low-Rank Adaptation of Large Language Models"☆10,327Updated last month
- Train transformer language models with reinforcement learning.☆9,288Updated this week
- Accessible large language models via k-bit quantization for PyTorch.☆6,029Updated this week
- Ongoing research training transformer models at scale☆9,949Updated this week
- Transformer related optimization, including BERT, GPT☆5,773Updated 5 months ago
- Hackable and optimized Transformers building blocks, supporting a composable construction.☆8,351Updated this week
- QLoRA: Efficient Finetuning of Quantized LLMs☆9,906Updated 3 months ago
- A high-throughput and memory-efficient inference and serving engine for LLMs☆26,822Updated this week
- RWKV is an RNN with transformer-level LLM performance. It can be directly trained like a GPT (parallelizable). So it's combining the best…☆12,397Updated 2 weeks ago
- A framework for few-shot evaluation of language models.☆6,426Updated this week
- Large-scale Self-supervised Pre-training Across Tasks, Languages, and Modalities☆19,545Updated 3 weeks ago
- Mamba SSM architecture☆12,542Updated last month
- LAVIS - A One-stop Library for Language-Vision Intelligence☆9,663Updated 3 weeks ago
- Latest Advances on Multimodal Large Language Models☆11,722Updated this week
- An open source implementation of CLIP.☆9,782Updated last month
- Large Language Model Text Generation Inference☆8,762Updated this week
- [NeurIPS'23 Oral] Visual Instruction Tuning (LLaVA) built towards GPT-4V level capabilities and beyond.☆19,294Updated last month
- Implementation of the LLaMA language model based on nanoGPT. Supports flash attention, Int8 and GPTQ 4bit quantization, LoRA and LLaMA-Ad…☆5,958Updated 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…☆8,186Updated last week
- Scripts for fine-tuning Meta Llama3 with composable FSDP & PEFT methods to cover single/multi-node GPUs. Supports default & custom datase…☆11,582Updated last week
- [ICLR 2024] Fine-tuning LLaMA to follow Instructions within 1 Hour and 1.2M Parameters☆5,691Updated 6 months ago
- Development repository for the Triton language and compiler☆12,698Updated this week
- [ICLR 2024] Efficient Streaming Language Models with Attention Sinks☆6,537Updated 2 months ago
- ImageBind One Embedding Space to Bind Them All☆8,221Updated last month
- ☆4,006Updated 3 months ago
- SGLang is a fast serving framework for large language models and vision language models.☆5,121Updated this week
- Example models using DeepSpeed☆5,987Updated this week
- A collection of libraries to optimise AI model performances☆8,373Updated last month