graphcore-research / unit-scaling-demo
Unit Scaling demo and experimentation code
☆16Updated 10 months ago
Alternatives and similar repositories for unit-scaling-demo:
Users that are interested in unit-scaling-demo are comparing it to the libraries listed below
- Repository for CPU Kernel Generation for LLM Inference☆25Updated last year
- Repository for Sparse Finetuning of LLMs via modified version of the MosaicML llmfoundry☆40Updated last year
- ☆58Updated 8 months ago
- Boosting 4-bit inference kernels with 2:4 Sparsity☆64Updated 4 months ago
- Flexible simulator for mixed precision and format simulation of LLMs and vision transformers.☆47Updated last year
- ACL 2023☆38Updated last year
- FlexAttention w/ FlashAttention3 Support☆27Updated 3 months ago
- Fast Hadamard transform in CUDA, with a PyTorch interface☆135Updated 8 months ago
- ☆22Updated last year
- Low-Rank Llama Custom Training☆21Updated 10 months ago
- PyTorch bindings for CUTLASS grouped GEMM.☆61Updated 3 months ago
- extensible collectives library in triton☆77Updated 4 months ago
- ☆21Updated 6 months ago
- ☆97Updated 5 months ago
- Odysseus: Playground of LLM Sequence Parallelism☆64Updated 7 months ago
- Here we will test various linear attention designs.☆58Updated 9 months ago
- NAACL '24 (Best Demo Paper RunnerUp) / MlSys @ NeurIPS '23 - RedCoast: A Lightweight Tool to Automate Distributed Training and Inference☆63Updated last month
- ☆23Updated 2 months ago
- Tritonbench is a collection of PyTorch custom operators with example inputs to measure their performance.☆86Updated this week
- 🔥 A minimal training framework for scaling FLA models☆32Updated this week
- QuIP quantization☆48Updated 10 months ago
- GPTQ inference TVM kernel☆38Updated 9 months ago
- Sirius, an efficient correction mechanism, which significantly boosts Contextual Sparsity models on reasoning tasks while maintaining its…☆21Updated 4 months ago
- Activation-aware Singular Value Decomposition for Compressing Large Language Models☆55Updated 3 months ago
- GPU operators for sparse tensor operations☆30Updated 10 months ago
- ☆27Updated 10 months ago
- ☆58Updated last week
- ☆17Updated last month
- [ACL 2024] RelayAttention for Efficient Large Language Model Serving with Long System Prompts☆38Updated 11 months ago
- Memory Optimizations for Deep Learning (ICML 2023)☆62Updated 10 months ago