hpcaitech / Elixir
Elixir: Train a Large Language Model on a Small GPU Cluster
☆12Updated last year
Related projects: ⓘ
- NAACL '24 (Best Demo Paper RunnerUp) / MlSys @ NeurIPS '23 - RedCoast: A Lightweight Tool to Automate Distributed Training and Inference☆58Updated this week
- Odysseus: Playground of LLM Sequence Parallelism☆50Updated 3 months ago
- ☆50Updated 3 months ago
- Repository for CPU Kernel Generation for LLM Inference☆25Updated last year
- A Suite for Parallel Inference of Diffusion Transformers (DiTs) on multi-GPU Clusters☆25Updated last month
- ☆83Updated 3 weeks ago
- (NeurIPS 2022) Automatically finding good model-parallel strategies, especially for complex models and clusters.☆33Updated last year
- Boosting 4-bit inference kernels with 2:4 Sparsity☆47Updated 2 weeks ago
- [ACL 2024] RelayAttention for Efficient Large Language Model Serving with Long System Prompts☆29Updated 6 months ago
- A Python library transfers PyTorch tensors between CPU and NVMe☆92Updated last year
- Summary of system papers/frameworks/codes/tools on training or serving large model☆56Updated 9 months ago
- Breaking Throughput-Latency Trade-off for Long Sequences with Speculative Decoding☆55Updated this week
- GEAR: An Efficient KV Cache Compression Recipefor Near-Lossless Generative Inference of LLM☆134Updated 2 months ago
- Repository for Sparse Finetuning of LLMs via modified version of the MosaicML llmfoundry☆36Updated 8 months ago
- Inference framework for MoE layers based on TensorRT with Python binding☆41Updated 3 years ago
- PyTorch library for cost-effective, fast and easy serving of MoE models.☆90Updated last month
- ☆35Updated 2 months ago
- A minimal implementation of vllm.☆29Updated last month
- PyTorch implementation of paper "Response Length Perception and Sequence Scheduling: An LLM-Empowered LLM Inference Pipeline".☆70Updated last year
- TensorRT LLM Benchmark Configuration☆10Updated last month
- Python package for rematerialization-aware gradient checkpointing☆22Updated 10 months ago
- GPTQ inference TVM kernel☆35Updated 4 months ago
- ☆75Updated this week
- ☆30Updated 3 months ago
- Triton implementation of Flash Attention2.0☆21Updated last year
- Ouroboros: Speculative Decoding with Large Model Enhanced Drafting☆60Updated 6 months ago
- ☆16Updated this week
- ☆38Updated 9 months ago
- ☆32Updated last week
- ☆65Updated 2 years ago