Step by step implementation of a fast softmax kernel in CUDA
☆62Jan 6, 2025Updated last year
Alternatives and similar repositories for FastSoftmax
Users that are interested in FastSoftmax are comparing it to the libraries listed below
Sorting:
- learn TensorRT from scratch🥰☆18Sep 29, 2024Updated last year
- A CUDA kernel for NHWC GroupNorm for PyTorch☆23Nov 15, 2024Updated last year
- Residual Quantization Autoencoder, used for interpreting LLMs☆14Jan 1, 2025Updated last year
- Standalone commandline CLI tool for compiling Triton kernels☆20Sep 13, 2024Updated last year
- Row-wise block scaling for fp8 quantization matrix multiplication. Solution to GPU mode AMD challenge.☆17Feb 9, 2026Updated 3 weeks ago
- Cute layout visualization☆30Jan 18, 2026Updated last month
- Flash Attention in ~100 lines of CUDA (forward pass only)☆11Jun 10, 2024Updated last year
- ☆15Feb 24, 2026Updated last week
- paper-read-notes☆13Sep 26, 2024Updated last year
- ☆14Nov 3, 2025Updated 4 months ago
- Implementation of a histogram equalization program using CUDA. Histogram equalization is a technique for adjusting image intensities to e…☆13Jan 3, 2021Updated 5 years ago
- Flash Attention in raw Cuda C beating PyTorch☆37May 14, 2024Updated last year
- TensorRT-in-Action 是一个 GitHub 代码库，提供了使用 TensorRT 的代码示例，并有对应 Jupyter Notebook。☆15Jun 1, 2023Updated 2 years ago
- 搜藏的希望的代码片段☆13Jun 6, 2023Updated 2 years ago
- Personal solutions to the Triton Puzzles☆20Jul 18, 2024Updated last year
- 使用mnn-llm对GOT-OCR2.0进行推理☆14Oct 2, 2024Updated last year
- Super fast FP32 matrix multiplication on RDNA3☆86Mar 30, 2025Updated 11 months ago
- segmentation algorithm yolact use tensorrt deploy☆14May 7, 2022Updated 3 years ago
- 一个轻量化的大模型推理框架☆21May 26, 2025Updated 9 months ago
- Official Problem Sets / Reference Kernels for the GPU MODE Leaderboard!☆212Updated this week
- 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
- ☆18Jan 4, 2024Updated 2 years ago
- HunyuanDiT with TensorRT and libtorch☆18May 22, 2024Updated last year
- Inference Llama 2 in one file of pure Cuda☆17Aug 20, 2023Updated 2 years ago
- 📚 A curated list of awesome matrix-matrix multiplication (A * B = C) frameworks, libraries and software☆61Feb 23, 2025Updated last year
- minimal C implementation of speculative decoding based on llama2.c☆25Jul 15, 2024Updated last year
- 高性能 高精度 大陆车牌、港澳车牌、台湾车牌 韩国车牌(South Korea LPR)识别 代码开源(ncnn移植）☆41Nov 5, 2025Updated 4 months ago
- ☆20Apr 29, 2016Updated 9 years ago
- Awesome code, projects, books, etc. related to CUDA☆31Feb 3, 2026Updated last month
- ☆20Dec 29, 2023Updated 2 years ago
- Using Rainbow implementation in Chainer RL for Slime Volleyball Pixel Environment☆23Jun 8, 2020Updated 5 years ago
- ☆26Nov 21, 2024Updated last year
- Fast CUDA matrix multiplication from scratch☆1,071Sep 2, 2025Updated 6 months ago
- Experimental GPU language with meta-programming☆26Sep 6, 2024Updated last year
- 车道线检测Lanenet TensorRT加速C++实现☆23Feb 24, 2022Updated 4 years ago
- Llama3 Streaming Chat Sample☆22Apr 24, 2024Updated last year
- ☆26Aug 15, 2023Updated 2 years ago
- 使用 CUDA C++ 实现的 llama 模型推理框架☆64Nov 8, 2024Updated last year
- ☆30Nov 16, 2024Updated last year