BlackSamorez / tensor_parallelLinks
Automatically split your PyTorch models on multiple GPUs for training & inference
☆657Updated 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:
- [ICML 2024] Break the Sequential Dependency of LLM Inference Using Lookahead Decoding☆1,288Updated 7 months ago
- Large Context Attention☆743Updated last week
- distributed trainer for LLMs☆581Updated last year
- Fast Inference Solutions for BLOOM☆565Updated last year
- [ICLR 2024] Sheared LLaMA: Accelerating Language Model Pre-training via Structured Pruning☆631Updated last year
- Microsoft Automatic Mixed Precision Library☆626Updated last year
- Memory optimization and training recipes to extrapolate language models' context length to 1 million tokens, with minimal hardware.☆747Updated last year
- [ICML 2024] SqueezeLLM: Dense-and-Sparse Quantization☆704Updated last year
- Official code for ReLoRA from the paper Stack More Layers Differently: High-Rank Training Through Low-Rank Updates☆465Updated last year
- Ongoing research training transformer language models at scale, including: BERT & GPT-2☆1,422Updated last year
- ☆546Updated 10 months ago
- Official PyTorch implementation of QA-LoRA☆141Updated last year
- Ring attention implementation with flash attention☆901Updated last month
- FP16xINT4 LLM inference kernel that can achieve near-ideal ~4x speedups up to medium batchsizes of 16-32 tokens.☆916Updated last year
- Explorations into some recent techniques surrounding speculative decoding☆288Updated 10 months ago
- Pipeline Parallelism for PyTorch☆780Updated last year
- Scalable toolkit for efficient model alignment☆841Updated 2 weeks ago
- Implementation of 💍 Ring Attention, from Liu et al. at Berkeley AI, in Pytorch☆542Updated 5 months ago
- For releasing code related to compression methods for transformers, accompanying our publications☆446Updated 9 months ago
- Code for the ICML 2023 paper "SparseGPT: Massive Language Models Can Be Accurately Pruned in One-Shot".☆843Updated last year
- Official repository of NEFTune: Noisy Embeddings Improves Instruction Finetuning☆401Updated last year
- Code for the ALiBi method for transformer language models (ICLR 2022)☆544Updated last year
- GPTQ inference Triton kernel☆311Updated 2 years ago
- Fast inference from large lauguage models via speculative decoding☆841Updated last year
- Code repo for the paper "LLM-QAT Data-Free Quantization Aware Training for Large Language Models"☆317Updated 7 months ago
- Best practice for training LLaMA models in Megatron-LM☆660Updated last year
- Serving multiple LoRA finetuned LLM as one☆1,101Updated last year
- Rectified Rotary Position Embeddings☆380Updated last year
- Code for the ICLR 2023 paper "GPTQ: Accurate Post-training Quantization of Generative Pretrained Transformers".☆2,202Updated last year
- LOMO: LOw-Memory Optimization☆988Updated last year