deepspeedai / DeepSpeed-KernelsLinks
☆74Updated 6 months ago
Alternatives and similar repositories for DeepSpeed-Kernels
Users that are interested in DeepSpeed-Kernels are comparing it to the libraries listed below
Sorting:
- ☆90Updated 10 months ago
- A Python library transfers PyTorch tensors between CPU and NVMe☆121Updated 10 months ago
- ☆112Updated last year
- ☆95Updated 6 months ago
- Boosting 4-bit inference kernels with 2:4 Sparsity☆82Updated last year
- Framework to reduce autotune overhead to zero for well known deployments.☆84Updated 2 weeks ago
- ☆98Updated 4 months ago
- extensible collectives library in triton☆88Updated 6 months ago
- ☆121Updated last year
- PyTorch bindings for CUTLASS grouped GEMM.☆123Updated 4 months ago
- An efficient GPU support for LLM inference with x-bit quantization (e.g. FP6,FP5).☆265Updated 2 months ago
- High-speed GEMV kernels, at most 2.7x speedup compared to pytorch baseline.☆116Updated last year
- TileFusion is an experimental C++ macro kernel template library that elevates the abstraction level in CUDA C for tile processing.☆97Updated 3 months ago
- Flash-LLM: Enabling Cost-Effective and Highly-Efficient Large Generative Model Inference with Unstructured Sparsity☆221Updated 2 years ago
- 🚀 Collection of components for development, training, tuning, and inference of foundation models leveraging PyTorch native components.☆213Updated last week
- DeeperGEMM: crazy optimized version☆71Updated 4 months ago
- ☆120Updated last year
- Odysseus: Playground of LLM Sequence Parallelism☆77Updated last year
- ☆64Updated 5 months ago
- Fast and memory-efficient exact attention☆95Updated this week
- [ICLR2025] Breaking Throughput-Latency Trade-off for Long Sequences with Speculative Decoding☆128Updated 10 months ago
- Applied AI experiments and examples for PyTorch☆296Updated last month
- Benchmark code for the "Online normalizer calculation for softmax" paper☆101Updated 7 years ago
- llama INT4 cuda inference with AWQ☆55Updated 8 months ago
- A Suite for Parallel Inference of Diffusion Transformers (DiTs) on multi-GPU Clusters☆50Updated last year
- ☆78Updated 10 months ago
- Standalone Flash Attention v2 kernel without libtorch dependency