BlackSamorez / tensor_parallel
Automatically split your PyTorch models on multiple GPUs for training & inference
☆652Updated last year
Alternatives and similar repositories for tensor_parallel
Users that are interested in tensor_parallel are comparing it to the libraries listed below
Sorting:
- Large Context Attention☆710Updated 3 months ago
- Ring attention implementation with flash attention☆764Updated last month
- Official code for ReLoRA from the paper Stack More Layers Differently: High-Rank Training Through Low-Rank Updates☆454Updated last year
- [ICML 2024] Break the Sequential Dependency of LLM Inference Using Lookahead Decoding☆1,248Updated 2 months ago
- Fast Inference Solutions for BLOOM☆561Updated 7 months ago
- FP16xINT4 LLM inference kernel that can achieve near-ideal ~4x speedups up to medium batchsizes of 16-32 tokens.☆819Updated 8 months ago
- train llama on a single A100 80G node using 🤗 transformers and 🚀 Deepspeed Pipeline Parallelism☆219Updated last year
- [ICLR 2024] Sheared LLaMA: Accelerating Language Model Pre-training via Structured Pruning☆608Updated last year
- Pipeline Parallelism for PyTorch☆765Updated 8 months ago
- Ongoing research training transformer language models at scale, including: BERT & GPT-2☆1,386Updated last year
- Implementation of 💍 Ring Attention, from Liu et al. at Berkeley AI, in Pytorch☆514Updated this week
- Microsoft Automatic Mixed Precision Library☆599Updated 7 months ago
- Official PyTorch implementation of QA-LoRA☆135Updated last year
- distributed trainer for LLMs☆575Updated 11 months ago
- Memory optimization and training recipes to extrapolate language models' context length to 1 million tokens, with minimal hardware.☆725Updated 7 months ago
- [ICML 2024] SqueezeLLM: Dense-and-Sparse Quantization☆688Updated 9 months ago
- Official repository of NEFTune: Noisy Embeddings Improves Instruction Finetuning☆395Updated last year
- GPTQ inference Triton kernel☆299Updated last year
- 📰 Must-read papers and blogs on Speculative Decoding ⚡️☆725Updated last week
- Scalable toolkit for efficient model alignment☆794Updated 2 weeks ago
- USP: Unified (a.k.a. Hybrid, 2D) Sequence Parallel Attention for Long Context Transformers Model Training and Inference☆492Updated 3 weeks ago
- REST: Retrieval-Based Speculative Decoding, NAACL 2024☆201Updated 5 months ago
- This repository contains code to quantitatively evaluate instruction-tuned models such as Alpaca and Flan-T5 on held-out tasks.☆547Updated last year
- Code for the ICLR 2023 paper "GPTQ: Accurate Post-training Quantization of Generative Pretrained Transformers".☆2,104Updated last year
- Fast inference from large lauguage models via speculative decoding☆723Updated 8 months ago
- ☆543Updated 5 months ago
- Code used for sourcing and cleaning the BigScience ROOTS corpus☆312Updated 2 years ago
- [NeurIPS'23] H2O: Heavy-Hitter Oracle for Efficient Generative Inference of Large Language Models.☆444Updated 9 months ago
- Official repository for LongChat and LongEval☆519Updated 11 months ago
- Multipack distributed sampler for fast padding-free training of LLMs☆189Updated 9 months ago