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:
- A CUDA kernel for NHWC GroupNorm for PyTorch☆26Nov 15, 2024Updated last year
- Tensara's GPU programming problems☆21Apr 23, 2026Updated 3 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
- ☆18Mar 12, 2025Updated last year
- Deploy on Railway without the complexity - Free Credits Offer • AdConnect your repo and Railway handles the rest with instant previews. Quickly provision container image services, databases, and storage volumes.
- 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☆96Nov 25, 2025Updated 8 months ago
- Row-wise block scaling for fp8 quantization matrix multiplication. Solution to GPU mode AMD challenge.☆19Feb 9, 2026Updated 6 months ago
- A collection of GPU experiments and benchmarks for my personal understanding and research.☆41Jul 22, 2026Updated 3 weeks ago
- Flash Attention in ~100 lines of CUDA (forward pass only)☆12Jun 10, 2024Updated 2 years ago
- ACPO: Python ML training/inference framework for LLVM optimizations☆13Jul 2, 2025Updated last year
- paper-read-notes☆13Sep 26, 2024Updated last year
- Standalone commandline CLI tool for compiling Triton kernels☆20Sep 13, 2024Updated last year
- Personal solutions to the Triton Puzzles☆21Jul 18, 2024Updated 2 years ago
- 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.
- Fast CUDA matrix multiplication from scratch☆1,280Sep 2, 2025Updated 11 months ago
- ☆14Nov 3, 2025Updated 9 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,…☆43May 17, 2026Updated 2 months ago
- TensorRT-in-Action 是一个 GitHub 代码库，提供了使用 TensorRT 的代码示例，并有对应 Jupyter Notebook。☆15Jun 1, 2023Updated 3 years ago
- 搜藏的希望的代码片段☆13Jun 6, 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 8 months ago
- Using Rainbow implementation in Chainer RL for Slime Volleyball Pixel Environment☆23Jun 8, 2020Updated 6 years ago
- Proton VPN Special Offer - Get 70% off • AdSpecial partner offer. Trusted by over 100 million users worldwide. Tested, Approved and Recommended by Experts.
- 使用mnn-llm对GOT-OCR2.0进行推理☆13Oct 2, 2024Updated last year
- ☆176Jul 29, 2026Updated 2 weeks ago
- Flash Attention in raw Cuda C beating PyTorch☆39May 14, 2024Updated 2 years ago
- WLAN channel access through Multi-Agent Reinforcement Learning (MARL)☆11Mar 2, 2022Updated 4 years ago
- This repository includes code to reproduce the tables in "Loss Landscapes are All You Need: Neural Network Generalization Can Be Explaine…☆40Mar 2, 2023Updated 3 years 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++.☆256Aug 6, 2026Updated last week
- 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.
- A lightweight, single-header C++11 Jinja2 template engine for LLM chat templates.☆20Mar 4, 2026Updated 5 months ago
- Fast and Memory-Efficient Exact Attention for Large Headdim, 1.5x~6x speedup over PyTorch SDPA.☆322Updated this week
- Inference Llama 2 in one file of pure Cuda☆17Aug 20, 2023Updated 2 years ago
- 一个轻量化的大模型推理框架☆23May 26, 2025Updated last year
- IMMA-based **FP8-as-storage** GEMM experiments for Ampere (sm_86 / RTX 3090 Ti).☆24Jan 30, 2026Updated 6 months ago
- ☆17Mar 29, 2026Updated 4 months ago
- ☆21Jun 12, 2026Updated 2 months ago