BlackSamorez / tensor_parallelLinks
Automatically split your PyTorch models on multiple GPUs for training & inference
☆656Updated 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,259Updated 4 months ago
- [ICLR 2024] Sheared LLaMA: Accelerating Language Model Pre-training via Structured Pruning☆618Updated last year
- Large Context Attention☆718Updated 5 months ago
- distributed trainer for LLMs☆578Updated last year
- Fast Inference Solutions for BLOOM☆564Updated 9 months ago
- Ring attention implementation with flash attention☆800Updated last week
- Memory optimization and training recipes to extrapolate language models' context length to 1 million tokens, with minimal hardware.☆734Updated 9 months ago
- [ICML 2024] SqueezeLLM: Dense-and-Sparse Quantization☆695Updated 10 months ago
- Official code for ReLoRA from the paper Stack More Layers Differently: High-Rank Training Through Low-Rank Updates☆458Updated last year
- Official PyTorch implementation of QA-LoRA☆138Updated last year
- Microsoft Automatic Mixed Precision Library☆612Updated 9 months ago
- ☆543Updated 6 months ago
- Scalable toolkit for efficient model alignment☆820Updated this week
- Ongoing research training transformer language models at scale, including: BERT & GPT-2☆1,401Updated last year
- FP16xINT4 LLM inference kernel that can achieve near-ideal ~4x speedups up to medium batchsizes of 16-32 tokens.☆852Updated 10 months ago
- Pipeline Parallelism for PyTorch☆769Updated 10 months ago
- Implementation of 💍 Ring Attention, from Liu et al. at Berkeley AI, in Pytorch☆526Updated last month
- Rectified Rotary Position Embeddings☆374Updated last year
- Explorations into some recent techniques surrounding speculative decoding☆272Updated 6 months ago
- Official repository of NEFTune: Noisy Embeddings Improves Instruction Finetuning☆396Updated last year
- LOMO: LOw-Memory Optimization☆987Updated last year
- Implementation of MEGABYTE, Predicting Million-byte Sequences with Multiscale Transformers, in Pytorch☆645Updated 6 months ago
- Code for the ALiBi method for transformer language models (ICLR 2022)☆536Updated last year
- Serving multiple LoRA finetuned LLM as one☆1,070Updated last year
- YaRN: Efficient Context Window Extension of Large Language Models☆1,511Updated last year
- [ICML 2023] SmoothQuant: Accurate and Efficient Post-Training Quantization for Large Language Models☆1,438Updated last year
- For releasing code related to compression methods for transformers, accompanying our publications☆433Updated 5 months ago
- train llama on a single A100 80G node using 🤗 transformers and 🚀 Deepspeed Pipeline Parallelism☆223Updated last year
- Code for the ICML 2023 paper "SparseGPT: Massive Language Models Can Be Accurately Pruned in One-Shot".☆812Updated 10 months ago
- Official Implementation of EAGLE-1 (ICML'24), EAGLE-2 (EMNLP'24), and EAGLE-3.☆1,371Updated last week