Dao-AILab / flash-attentionLinks
Fast and memory-efficient exact attention
☆17,846Updated this week
Alternatives and similar repositories for flash-attention
Users that are interested in flash-attention are comparing it to the libraries listed below
Sorting:
- Accessible large language models via k-bit quantization for PyTorch.☆7,142Updated this week
- Hackable and optimized Transformers building blocks, supporting a composable construction.☆9,591Updated last week
- 🚀 A simple way to launch, train, and use PyTorch models on almost any device and distributed configuration, automatic mixed precision (i…☆8,839Updated this week
- 🤗 PEFT: State-of-the-art Parameter-Efficient Fine-Tuning.☆18,774Updated last week
- Transformer related optimization, including BERT, GPT☆6,211Updated last year
- Train transformer language models with reinforcement learning.☆14,193Updated this week
- Ongoing research training transformer models at scale☆12,600Updated this week
- An open source implementation of CLIP.☆11,957Updated last week
- Large Language Model Text Generation Inference☆10,236Updated this week
- A high-throughput and memory-efficient inference and serving engine for LLMs☆49,721Updated this week
- [NeurIPS'23 Oral] Visual Instruction Tuning (LLaVA) built towards GPT-4V level capabilities and beyond.☆22,795Updated 10 months ago
- QLoRA: Efficient Finetuning of Quantized LLMs☆10,500Updated last year
- DeepSpeed is a deep learning optimization library that makes distributed training and inference easy, efficient, and effective.☆38,997Updated this week
- A framework for few-shot evaluation of language models.☆9,326Updated this week
- TensorRT-LLM provides users with an easy-to-use Python API to define Large Language Models (LLMs) and support state-of-the-art optimizati…☆10,734Updated this week
- An easy-to-use LLMs quantization package with user-friendly apis, based on GPTQ algorithm.☆4,873Updated 2 months ago
- Code for loralib, an implementation of "LoRA: Low-Rank Adaptation of Large Language Models"☆12,090Updated 6 months ago
- PyTorch native post-training library☆5,273Updated this week
- SGLang is a fast serving framework for large language models and vision language models.☆15,276Updated this week
- LMDeploy is a toolkit for compressing, deploying, and serving LLMs.☆6,517Updated last week
- Welcome to the Llama Cookbook! This is your go to guide for Building with Llama: Getting started with Inference, Fine-Tuning, RAG. We als…☆17,490Updated this week
- Large-scale Self-supervised Pre-training Across Tasks, Languages, and Modalities☆21,420Updated 2 weeks ago
- The official repo of Qwen-VL (通义千问-VL) chat & pretrained large vision language model proposed by Alibaba Cloud.☆5,993Updated 10 months ago
- LAVIS - A One-stop Library for Language-Vision Intelligence☆10,636Updated 7 months ago
- [MLSys 2024 Best Paper Award] AWQ: Activation-aware Weight Quantization for LLM Compression and Acceleration☆3,081Updated last week
- Simple and efficient pytorch-native transformer text generation in <1000 LOC of python.☆5,993Updated 2 months ago
- PyTorch extensions for high performance and large scale training.☆3,331Updated last month
- a state-of-the-art-level open visual language model | 多模态预训练模型☆6,589Updated last year
- Tensor library for machine learning☆12,697Updated last week
- CLIP (Contrastive Language-Image Pretraining), Predict the most relevant text snippet given an image☆29,407Updated 10 months ago