Step by step implementation of a fast softmax kernel in CUDA
☆71Jan 6, 2025Updated last year
Alternatives and similar repositories for FastSoftmax
Users that are interested in FastSoftmax are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- ☆101May 30, 2026Updated 3 months ago
- A CUDA kernel for NHWC GroupNorm for PyTorch☆26Nov 15, 2024Updated last year
- Tensara's GPU programming problems☆21Apr 23, 2026Updated 5 months ago
- learn TensorRT from scratch🥰☆18Sep 29, 2024Updated last year
- My submission for the GPUMODE/AMD fp8 mm challenge☆29Jun 4, 2025Updated 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.
- ☆18Mar 12, 2025Updated last year
- Decoding Attention is specially optimized for MHA, MQA, GQA and MLA using CUDA core for the decoding stage of LLM inference.☆48Jun 11, 2025Updated last year
- Hand-Rolled GPU communications library☆94Nov 25, 2025Updated 10 months ago
- Row-wise block scaling for fp8 quantization matrix multiplication. Solution to GPU mode AMD challenge.☆19Feb 9, 2026Updated 7 months ago
- A collection of GPU experiments and benchmarks for my personal understanding and research.☆43Aug 19, 2026Updated last month
- A simple docker compose setup that works (in limited testing) on Blackwell cards☆16Oct 13, 2025Updated 11 months ago
- Flash Attention in ~100 lines of CUDA (forward pass only)☆12Jun 10, 2024Updated 2 years ago
- Standalone commandline CLI tool for compiling Triton kernels☆20Sep 13, 2024Updated 2 years ago
- Personal solutions to the Triton Puzzles☆22Jul 18, 2024Updated 2 years ago
- AI Agents on DigitalOcean Gradient AI Platform • AdBuild production-ready AI agents using customizable tools or access multiple LLMs through a single endpoint. Create custom knowledge bases or connect external data.
- Fast CUDA matrix multiplication from scratch☆1,317Sep 2, 2025Updated last year
- ACPO: Python ML training/inference framework for LLVM optimizations☆15Jul 2, 2025Updated last year
- ☆14Nov 3, 2025Updated 10 months ago
- Super fast FP32 matrix multiplication on RDNA3☆92Mar 30, 2025Updated last year
- A GPT-2 inference engine written from scratch in CUDA and C++. Implements custom CUDA kernels for tiled matrix multiplication, LayerNorm,…☆44May 17, 2026Updated 4 months ago
- TensorRT-in-Action 是一个 GitHub 代码库，提供了使用 TensorRT 的代码示例，并有对应 Jupyter Notebook。☆15Jun 1, 2023Updated 3 years ago
- HunyuanDiT with TensorRT and libtorch☆18May 22, 2024Updated 2 years ago
- a ai infra framework for edge device base on nndeploy☆18Nov 27, 2025Updated 9 months ago
- 使用mnn-llm对GOT-OCR2.0进行推理☆13Oct 2, 2024Updated last year
- Managed hosting for WordPress and PHP on Cloudways • AdManaged hosting for WordPress, Magento, Laravel, or PHP apps, on multiple cloud providers. Deploy in minutes on Cloudways by DigitalOcean.
- ☆182Sep 18, 2026Updated last week
- Flash Attention in raw Cuda C beating PyTorch☆39May 14, 2024Updated 2 years ago
- A high-performance attention mechanism that computes softmax normalization in a single streaming pass using running accumulators (online …☆32Sep 9, 2026Updated 2 weeks ago
- ☆19Jun 6, 2025Updated last year
- segmentation algorithm yolact use tensorrt deploy☆14May 7, 2022Updated 4 years ago
- ☆19Jan 4, 2024Updated 2 years ago
- Quantized LLM training in pure CUDA/C++.☆257Aug 31, 2026Updated 3 weeks ago
- A lightweight, single-header C++11 Jinja2 template engine for LLM chat templates.☆20Mar 4, 2026Updated 6 months ago
- Kernel Library for Large Headdim Attention (64~1024, BF16/FP8/FP4), 1.5x~15x↑ vs PyTorch SDPA.☆334Updated this week
- Virtual machines for every use case on DigitalOcean • AdGet dependable uptime with 99.99% SLA, simple security tools, and predictable monthly pricing with DigitalOcean's virtual machines, called Droplets.
- arkit demo☆11Aug 20, 2018Updated 8 years ago
- 一个轻量化的大模型推理框架☆23May 26, 2025Updated last year
- ☆17Mar 29, 2026Updated 5 months ago
- IMMA-based **FP8-as-storage** GEMM experiments for Ampere (sm_86 / RTX 3090 Ti).☆26Jan 30, 2026Updated 7 months ago
- ☆272Jul 11, 2024Updated 2 years ago
- ☆21Jun 12, 2026Updated 3 months ago
- A set of Python scripts that makes your experience on TPU better☆56Sep 18, 2025Updated last year