Benchmark code for the "Online normalizer calculation for softmax" paper
☆108Jul 27, 2018Updated 7 years ago
Alternatives and similar repositories for online-softmax
Users that are interested in online-softmax are comparing it to the libraries listed below
Sorting:
- A CUDA kernel for NHWC GroupNorm for PyTorch☆23Nov 15, 2024Updated last year
- NVSHMEM‑Tutorial: Build a DeepEP‑like GPU Buffer☆163Feb 11, 2026Updated 3 weeks ago
- Standalone Flash Attention v2 kernel without libtorch dependency☆114Sep 10, 2024Updated last year
- ☆262Jul 11, 2024Updated last year
- ☆21Mar 3, 2025Updated last year
- We invite you to visit and follow our new repository at https://github.com/microsoft/TileFusion. TiledCUDA is a highly efficient kernel …☆194Jan 28, 2025Updated last year
- ☆152Jan 9, 2025Updated last year
- ☆62Feb 15, 2026Updated 2 weeks ago
- ☆34Feb 3, 2025Updated last year
- A standalone GEMM kernel for fp16 activation and quantized weight, extracted from FasterTransformer☆96Feb 20, 2026Updated last week
- row-major matmul optimization☆707Feb 24, 2026Updated last week
- Several optimization methods of half-precision general matrix multiplication (HGEMM) using tensor core with WMMA API and MMA PTX instruct…☆526Sep 8, 2024Updated last year
- A simple high performance CUDA GEMM implementation.☆426Jan 4, 2024Updated 2 years ago
- Yinghan's Code Sample☆365Jul 25, 2022Updated 3 years ago
- Rust crate for some audio utilities☆27Mar 8, 2025Updated 11 months ago
- Inference Llama 2 in one file of pure Cuda☆17Aug 20, 2023Updated 2 years ago
- TASU: A New Style of Alignment of Speech LLM with only Text Training Data, zero-shot on ASR and Other SU tasks☆22Jan 19, 2026Updated last month
- 分层解耦的深度学习推理引擎☆79Feb 17, 2025Updated last year
- ☆141Apr 23, 2024Updated last year
- ⚡️Write HGEMM from scratch using Tensor Cores with WMMA, MMA and CuTe API, Achieve Peak⚡️ Performance.☆147May 10, 2025Updated 9 months ago
- ☆120Apr 2, 2025Updated 11 months ago
- PyTorch bindings for CUTLASS grouped GEMM.☆144May 29, 2025Updated 9 months ago
- Prototype routines for GPU quantization written using PyTorch.☆21Feb 8, 2026Updated 3 weeks ago
- Penn CIS 5650 (GPU Programming and Architecture) Final Project☆44Dec 11, 2023Updated 2 years ago
- ☆49Apr 15, 2024Updated last year
- Dynamic Memory Management for Serving LLMs without PagedAttention☆464May 30, 2025Updated 9 months ago
- Several optimization methods of half-precision general matrix vector multiplication (HGEMV) using CUDA core.☆73Sep 8, 2024Updated last year
- flash attention tutorial written in python, triton, cuda, cutlass☆488Jan 20, 2026Updated last month
- Decoding Attention is specially optimized for MHA, MQA, GQA and MLA using CUDA core for the decoding stage of LLM inference.☆46Jun 11, 2025Updated 8 months ago
- ☆123Feb 24, 2026Updated last week
- Distributed Compiler based on Triton for Parallel Systems☆1,371Feb 13, 2026Updated 2 weeks ago
- trt-hackathon-2022 三等奖方案☆10Mar 6, 2023Updated 2 years ago
- ☆10Jul 18, 2024Updated last year
- Inference deployment of the llama3☆11Apr 21, 2024Updated last year
- semantic tokenizer for speech and music☆21Jul 6, 2025Updated 7 months ago
- Framework for Algorithmic Correctness Testing of Operators☆16Feb 20, 2026Updated last week
- a fast and customizable CUDA int4 tensor core gemm☆15Aug 2, 2024Updated last year
- Integration test of Verilog AXI modules (https://github.com/alexforencich/verilog-axi) with LiteX.☆17Dec 19, 2022Updated 3 years ago
- ☆22Nov 7, 2018Updated 7 years ago