BlackSamorez / tensor_parallel
Automatically split your PyTorch models on multiple GPUs for training & inference
☆649Updated last year
Alternatives and similar repositories for tensor_parallel:
Users that are interested in tensor_parallel are comparing it to the libraries listed below
- [ICML 2024] Break the Sequential Dependency of LLM Inference Using Lookahead Decoding☆1,194Updated 4 months ago
- [ICLR 2024] Sheared LLaMA: Accelerating Language Model Pre-training via Structured Pruning☆585Updated 11 months ago
- Ongoing research training transformer language models at scale, including: BERT & GPT-2☆1,364Updated 11 months ago
- Large Context Attention☆684Updated 3 weeks ago
- [ICML 2024] SqueezeLLM: Dense-and-Sparse Quantization☆676Updated 6 months ago
- Pipeline Parallelism for PyTorch☆748Updated 6 months ago
- Ring attention implementation with flash attention☆677Updated this week
- Microsoft Automatic Mixed Precision Library☆567Updated 4 months ago
- FP16xINT4 LLM inference kernel that can achieve near-ideal ~4x speedups up to medium batchsizes of 16-32 tokens.☆729Updated 5 months ago
- distributed trainer for LLMs☆557Updated 9 months ago
- Scalable toolkit for efficient model alignment☆722Updated this week
- Memory optimization and training recipes to extrapolate language models' context length to 1 million tokens, with minimal hardware.☆700Updated 4 months ago
- Official Implementation of EAGLE-1 (ICML'24) and EAGLE-2 (EMNLP'24)☆968Updated this week
- USP: Unified (a.k.a. Hybrid, 2D) Sequence Parallel Attention for Long Context Transformers Model Training and Inference☆428Updated this week
- ☆539Updated 2 months ago
- Implementation of 💍 Ring Attention, from Liu et al. at Berkeley AI, in Pytorch☆502Updated 3 months ago
- ☆314Updated 10 months ago
- Serving multiple LoRA finetuned LLM as one☆1,029Updated 9 months ago
- Official code for ReLoRA from the paper Stack More Layers Differently: High-Rank Training Through Low-Rank Updates☆443Updated 10 months ago
- [ICML 2023] SmoothQuant: Accurate and Efficient Post-Training Quantization for Large Language Models☆1,341Updated 7 months ago
- Explorations into some recent techniques surrounding speculative decoding☆240Updated last month
- Fast inference from large lauguage models via speculative decoding☆661Updated 6 months ago
- train llama on a single A100 80G node using 🤗 transformers and 🚀 Deepspeed Pipeline Parallelism☆215Updated last year
- 📰 Must-read papers and blogs on Speculative Decoding ⚡️☆597Updated this week
- Fast Inference Solutions for BLOOM☆563Updated 4 months ago
- 🚀 Efficiently (pre)training foundation models with native PyTorch features, including FSDP for training and SDPA implementation of Flash…☆222Updated this week
- A simple and effective LLM pruning approach.☆712Updated 6 months ago
- Official repository for LongChat and LongEval☆519Updated 8 months ago
- [ICLR2024 spotlight] OmniQuant is a simple and powerful quantization technique for LLMs.☆767Updated 4 months ago
- Minimalistic large language model 3D-parallelism training☆1,483Updated this week