BlackSamorez / tensor_parallelLinks
Automatically split your PyTorch models on multiple GPUs for training & inference
☆656Updated 2 years ago
Alternatives and similar repositories for tensor_parallel
Users that are interested in tensor_parallel are comparing it to the libraries listed below
Sorting:
- [ICML 2024] Break the Sequential Dependency of LLM Inference Using Lookahead Decoding☆1,316Updated 11 months ago
- [ICLR 2024] Sheared LLaMA: Accelerating Language Model Pre-training via Structured Pruning☆639Updated last year
- Large Context Attention☆766Updated 3 months ago
- Fast Inference Solutions for BLOOM☆566Updated last year
- distributed trainer for LLMs☆588Updated last year
- ☆553Updated last year
- Official code for ReLoRA from the paper Stack More Layers Differently: High-Rank Training Through Low-Rank Updates☆473Updated last year
- Official PyTorch implementation of QA-LoRA☆145Updated last year
- Microsoft Automatic Mixed Precision Library☆635Updated 2 months ago
- Ongoing research training transformer language models at scale, including: BERT & GPT-2☆1,431Updated last year
- Memory optimization and training recipes to extrapolate language models' context length to 1 million tokens, with minimal hardware.☆751Updated last year
- [ICML 2024] SqueezeLLM: Dense-and-Sparse Quantization☆713Updated last year
- Code for the ICLR 2023 paper "GPTQ: Accurate Post-training Quantization of Generative Pretrained Transformers".☆2,249Updated last year
- Scalable toolkit for efficient model alignment☆852Updated 4 months ago
- Code for the ICML 2023 paper "SparseGPT: Massive Language Models Can Be Accurately Pruned in One-Shot".☆866Updated last year
- LOMO: LOw-Memory Optimization☆987Updated last year
- Pipeline Parallelism for PyTorch☆784Updated last year
- Ring attention implementation with flash attention☆979Updated 4 months ago
- A simple and effective LLM pruning approach.☆847Updated last year
- Central place for the engineering/scaling WG: documentation, SLURM scripts and logs, compute environment and data.☆1,008Updated last year
- Code for the ALiBi method for transformer language models (ICLR 2022)☆549Updated 2 years ago
- [NeurIPS 2023] LLM-Pruner: On the Structural Pruning of Large Language Models. Support Llama-3/3.1, Llama-2, LLaMA, BLOOM, Vicuna, Baich…☆1,105Updated last year
- train llama on a single A100 80G node using 🤗 transformers and 🚀 Deepspeed Pipeline Parallelism☆224Updated 2 years ago
- [ICML 2023] SmoothQuant: Accurate and Efficient Post-Training Quantization for Large Language Models☆1,600Updated last year
- Explorations into some recent techniques surrounding speculative decoding☆299Updated last year
- Implementation of MEGABYTE, Predicting Million-byte Sequences with Multiscale Transformers, in Pytorch☆654Updated last year
- Implementation of 💍 Ring Attention, from Liu et al. at Berkeley AI, in Pytorch☆549Updated 8 months ago
- Ongoing research training transformer language models at scale, including: BERT & GPT-2☆2,224Updated 5 months ago
- GPTQ inference Triton kernel☆321Updated 2 years ago
- Tutel MoE: Optimized Mixture-of-Experts Library, Support GptOss/DeepSeek/Kimi-K2/Qwen3 using FP8/NVFP4/MXFP4☆962Updated last month