☆75Mar 26, 2025Updated last year
Alternatives and similar repositories for DeepSpeed-Kernels
Users that are interested in DeepSpeed-Kernels are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- A standalone GEMM kernel for fp16 activation and quantized weight, extracted from FasterTransformer☆96Feb 20, 2026Updated 3 months ago
- MII makes low-latency and high-throughput inference possible, powered by DeepSpeed.☆2,109Jun 30, 2025Updated 11 months ago
- Performance of the C++ interface of flash attention and flash attention v2 in large language model (LLM) inference scenarios.☆45Feb 27, 2025Updated last year
- High-speed GEMV kernels, at most 2.7x speedup compared to pytorch baseline.☆129Jul 13, 2024Updated last year
- TileFusion is an experimental C++ macro kernel template library that elevates the abstraction level in CUDA C for tile processing.☆110Jun 28, 2025Updated 11 months ago
- Deploy to Railway using AI coding agents - Free Credits Offer • AdUse Claude Code, Codex, OpenCode, and more. Autonomous software development now has the infrastructure to match with Railway.
- Official resporitory for "IPDPS' 24 QSync: Quantization-Minimized Synchronous Distributed Training Across Hybrid Devices".☆20Feb 23, 2024Updated 2 years ago
- ☆25Mar 15, 2023Updated 3 years ago
- ☆267Jul 11, 2024Updated last year
- ☆20Sep 28, 2024Updated last year
- Perplexity GPU Kernels☆587Nov 7, 2025Updated 7 months ago
- Decoding Attention is specially optimized for MHA, MQA, GQA and MLA using CUDA core for the decoding stage of LLM inference.☆47Jun 11, 2025Updated 11 months ago
- MSCCL++: A GPU-driven communication stack for scalable AI applications☆530Updated this week
- ☆13Jan 7, 2025Updated last year
- CUDA 12.2 HMM demos☆21Jul 26, 2024Updated last year
- Managed Database hosting by DigitalOcean • AdPostgreSQL, MySQL, MongoDB, Kafka, Valkey, and OpenSearch available. Automatically scale up storage and focus on building your apps.
- A low-latency & high-throughput serving engine for LLMs☆506Jan 8, 2026Updated 5 months ago
- FP16xINT4 LLM inference kernel that can achieve near-ideal ~4x speedups up to medium batchsizes of 16-32 tokens.☆1,082Sep 4, 2024Updated last year
- IntLLaMA: A fast and light quantization solution for LLaMA☆18Jul 21, 2023Updated 2 years ago
- ☆134Nov 11, 2024Updated last year
- DeepGEMM: clean and efficient FP8 GEMM kernels with fine-grained scaling☆24Updated this week
- Dynamic Memory Management for Serving LLMs without PagedAttention☆487May 30, 2025Updated last year
- Implementation for IceFormer: Accelerated Inference with Long-Sequence Transformers on CPUs (ICLR 2024).☆25Feb 22, 2026Updated 3 months ago
- Sample Codes using NVSHMEM on Multi-GPU☆30Jan 22, 2023Updated 3 years ago
- GPTQ inference Triton kernel☆322May 18, 2023Updated 3 years ago
- Managed Kubernetes at scale on DigitalOcean • AdDigitalOcean Kubernetes includes the control plane, bandwidth allowance, container registry, automatic updates, and more for free.
- ☆106Sep 9, 2024Updated last year
- ☆26Feb 20, 2024Updated 2 years ago
- [MLSys'24] Atom: Low-bit Quantization for Efficient and Accurate LLM Serving☆341Jul 2, 2024Updated last year
- ⚡️Write HGEMM from scratch using Tensor Cores with WMMA, MMA and CuTe API, Achieve Peak⚡️ Performance.☆155May 10, 2025Updated last year
- ☆11Apr 5, 2021Updated 5 years ago
- Tritonbench is a collection of PyTorch custom operators with example inputs to measure their performance.☆358Jun 2, 2026Updated last week
- extensible collectives library in triton☆98Mar 31, 2025Updated last year
- Stateful LLM Serving☆102Mar 11, 2025Updated last year
- Expert Specialization MoE Solution based on CUTLASS☆27Apr 14, 2026Updated last month
- Managed Kubernetes at scale on DigitalOcean • AdDigitalOcean Kubernetes includes the control plane, bandwidth allowance, container registry, automatic updates, and more for free.
- optimized BERT transformer inference on NVIDIA GPU. https://arxiv.org/abs/2210.03052☆479Mar 15, 2024Updated 2 years ago
- Standalone Flash Attention v2 kernel without libtorch dependency☆113Sep 10, 2024Updated last year
- ☆113Mar 12, 2026Updated 2 months ago
- Torch Distributed Experimental☆117Aug 5, 2024Updated last year
- Microsoft Collective Communication Library☆390Sep 20, 2023Updated 2 years ago
- Depict GPU memory footprint during DNN training of PyTorch☆11Nov 17, 2022Updated 3 years ago
- A Easy-to-understand TensorOp Matmul Tutorial☆440Mar 5, 2026Updated 3 months ago