deepspeedai / DeepSpeed-Kernels
☆62Updated 3 weeks ago
Alternatives and similar repositories for DeepSpeed-Kernels:
Users that are interested in DeepSpeed-Kernels are comparing it to the libraries listed below
- PyTorch bindings for CUTLASS grouped GEMM.☆74Updated 4 months ago
- Boosting 4-bit inference kernels with 2:4 Sparsity☆71Updated 6 months ago
- ☆101Updated 6 months ago
- High-speed GEMV kernels, at most 2.7x speedup compared to pytorch baseline.☆101Updated 8 months ago
- Tritonbench is a collection of PyTorch custom operators with example inputs to measure their performance.☆104Updated this week
- extensible collectives library in triton☆84Updated 6 months ago
- Odysseus: Playground of LLM Sequence Parallelism☆66Updated 9 months ago
- TileFusion is an experimental C++ macro kernel template library that elevates the abstraction level in CUDA C for tile processing. By pro…☆68Updated this week
- ☆88Updated 6 months ago
- ☆73Updated 4 months ago
- A Python library transfers PyTorch tensors between CPU and NVMe☆111Updated 3 months ago
- Standalone Flash Attention v2 kernel without libtorch dependency☆106Updated 6 months ago
- [ICLR2025] Breaking Throughput-Latency Trade-off for Long Sequences with Speculative Decoding☆110Updated 3 months ago
- ☆64Updated 2 months ago
- An efficient GPU support for LLM inference with x-bit quantization (e.g. FP6,FP5).☆240Updated 4 months ago
- Framework to reduce autotune overhead to zero for well known deployments.☆63Updated this week
- A safetensors extension to efficiently store sparse quantized tensors on disk☆91Updated this week
- ☆116Updated last year
- Memory Optimizations for Deep Learning (ICML 2023)☆62Updated last year
- ☆55Updated 2 months ago
- ☆26Updated this week
- Fast low-bit matmul kernels in Triton☆267Updated this week
- Benchmark code for the "Online normalizer calculation for softmax" paper☆85Updated 6 years ago
- ☆38Updated last year
- ☆74Updated 3 months ago
- Fast and memory-efficient exact attention☆52Updated this week
- llama INT4 cuda inference with AWQ☆53Updated 2 months ago
- ☆157Updated last year
- ☆69Updated 3 months ago
- ☆25Updated this week