Aleph-Alpha / NeurIPS-WANT-submission-efficient-parallelization-layouts
☆22Updated last year
Alternatives and similar repositories for NeurIPS-WANT-submission-efficient-parallelization-layouts:
Users that are interested in NeurIPS-WANT-submission-efficient-parallelization-layouts are comparing it to the libraries listed below
- Odysseus: Playground of LLM Sequence Parallelism☆66Updated 9 months ago
- PyTorch bindings for CUTLASS grouped GEMM.☆73Updated 4 months ago
- ☆30Updated 9 months ago
- Transformers components but in Triton☆32Updated 4 months ago
- Sequence-level 1F1B schedule for LLMs.☆17Updated 9 months ago
- Quantized Attention on GPU☆45Updated 3 months ago
- Vocabulary Parallelism☆17Updated last week
- ☆38Updated last year
- GPU operators for sparse tensor operations☆30Updated last year
- GPTQ inference TVM kernel☆39Updated 10 months ago
- ☆19Updated last month
- Contextual Position Encoding but with some custom CUDA Kernels https://arxiv.org/abs/2405.18719☆22Updated 9 months ago
- Repository of the paper "Accelerating Transformer Inference for Translation via Parallel Decoding"☆114Updated last year
- Boosting 4-bit inference kernels with 2:4 Sparsity☆68Updated 6 months ago
- NAACL '24 (Best Demo Paper RunnerUp) / MlSys @ NeurIPS '23 - RedCoast: A Lightweight Tool to Automate Distributed Training and Inference☆64Updated 3 months ago
- Inference framework for MoE layers based on TensorRT with Python binding☆41Updated 3 years ago
- [ICLR 2025] TidalDecode: A Fast and Accurate LLM Decoding with Position Persistent Sparse Attention☆29Updated 2 weeks ago
- [NeurIPS 2024] Fast Best-of-N Decoding via Speculative Rejection☆39Updated 4 months ago
- 🔥 A minimal training framework for scaling FLA models☆79Updated this week
- [ICLR 2025] PEARL: parallel speculative decoding with adaptive draft length☆54Updated last week
- Ouroboros: Speculative Decoding with Large Model Enhanced Drafting (EMNLP 2024 main)☆86Updated 5 months ago
- Tritonbench is a collection of PyTorch custom operators with example inputs to measure their performance.☆104Updated this week
- Awesome Triton Resources☆20Updated 3 months ago
- SQUEEZED ATTENTION: Accelerating Long Prompt LLM Inference☆44Updated 3 months ago
- ☆14Updated 2 years ago
- 16-fold memory access reduction with nearly no loss☆81Updated 3 weeks ago
- Framework to reduce autotune overhead to zero for well known deployments.☆63Updated this week
- The official implementation of paper: SimLayerKV: A Simple Framework for Layer-Level KV Cache Reduction.☆43Updated 5 months ago