InternLM / turbomind
☆52Updated last month
Alternatives and similar repositories for turbomind:
Users that are interested in turbomind are comparing it to the libraries listed below
- Odysseus: Playground of LLM Sequence Parallelism☆64Updated 7 months ago
- PyTorch bindings for CUTLASS grouped GEMM.☆58Updated 2 months ago
- Tritonbench is a collection of PyTorch custom operators with example inputs to measure their performance.☆75Updated this week
- ☆78Updated 4 months ago
- Implement Flash Attention using Cute.☆65Updated last month
- ⚡️Write HGEMM from scratch using Tensor Cores with WMMA, MMA and CuTe API, achieve peak⚡️ performance☆43Updated this week
- A sparse attention kernel supporting mix sparse patterns☆93Updated 3 months ago
- 使用 cutlass 仓库在 ada 架构上实现 fp8 的 flash attention☆52Updated 5 months ago
- 📚[WIP] FFPA: Yet antother Faster Flash Prefill Attention with O(1)🎉GPU SRAM complexity for headdim > 256, 1.8x~3x↑🎉faster vs SDPA EA.☆44Updated this week
- Boosting 4-bit inference kernels with 2:4 Sparsity☆64Updated 4 months ago
- Materials for learning SGLang☆166Updated last week
- QQQ is an innovative and hardware-optimized W4A8 quantization solution for LLMs.☆94Updated last month
- Quantized Attention on GPU☆34Updated last month
- Standalone Flash Attention v2 kernel without libtorch dependency☆99Updated 4 months ago
- ☆59Updated last month
- Summary of system papers/frameworks/codes/tools on training or serving large model☆56Updated last year
- PyTorch bindings for CUTLASS grouped GEMM.☆84Updated 2 weeks ago
- ☆71Updated 5 months ago
- GPTQ inference TVM kernel☆38Updated 8 months ago
- A Suite for Parallel Inference of Diffusion Transformers (DiTs) on multi-GPU Clusters☆37Updated 5 months ago
- ☆55Updated 3 months ago
- [MLSys'24] Atom: Low-bit Quantization for Efficient and Accurate LLM Serving☆290Updated 6 months ago
- Performance of the C++ interface of flash attention and flash attention v2 in large language model (LLM) inference scenarios.☆33Updated 4 months ago
- Decoding Attention is specially optimized for multi head attention (MHA) using CUDA core for the decoding stage of LLM inference.☆27Updated 2 months ago
- An efficient GPU support for LLM inference with x-bit quantization (e.g. FP6,FP5).☆230Updated 2 months ago
- Official PyTorch implementation of FlatQuant: Flatness Matters for LLM Quantization☆90Updated 2 months ago
- Breaking Throughput-Latency Trade-off for Long Sequences with Speculative Decoding☆107Updated last month
- A standalone GEMM kernel for fp16 activation and quantized weight, extracted from FasterTransformer☆87Updated 10 months ago
- An easy-to-use package for implementing SmoothQuant for LLMs☆89Updated 8 months ago
- High-speed GEMV kernels, at most 2.7x speedup compared to pytorch baseline.☆93Updated 6 months ago