Step by step implementation of a fast softmax kernel in CUDA
☆72Jan 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:
- ☆100May 30, 2026Updated 4 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 2 years ago
- My submission for the GPUMODE/AMD fp8 mm challenge☆29Jun 4, 2025Updated last year
- Serverless GPU API endpoints on Runpod - Get Bonus Credits • AdSkip the infrastructure headaches. Auto-scaling, pay-as-you-go, no-ops approach lets you focus on innovating your application.
- ☆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☆95Nov 25, 2025Updated 10 months ago
- Row-wise block scaling for fp8 quantization matrix multiplication. Solution to GPU mode AMD challenge.☆19Feb 9, 2026Updated 8 months ago
- A collection of GPU experiments and benchmarks for my personal understanding and research.☆43Aug 19, 2026Updated last month
- Flash Attention in ~100 lines of CUDA (forward pass only)☆12Jun 10, 2024Updated 2 years ago
- paper-read-notes☆13Sep 26, 2024Updated 2 years ago
- DeepSeek R1 distilled into smaller OSS models for hobbyist☆17Dec 2, 2025Updated 10 months ago
- Standalone commandline CLI tool for compiling Triton kernels☆20Sep 13, 2024Updated 2 years ago
- 1-Click AI Models by DigitalOcean Gradient • AdDeploy popular AI models on DigitalOcean Gradient GPU virtual machines with just a single click. Zero configuration with optimized deployments.
- ACPO: Python ML training/inference framework for LLVM optimizations☆15Jul 2, 2025Updated last year
- Fast CUDA matrix multiplication from scratch☆1,335Sep 2, 2025Updated last year
- ☆14Nov 3, 2025Updated 11 months ago
- Personal solutions to the Triton Puzzles☆22Jul 18, 2024Updated 2 years ago
- A GPT-2 inference engine written from scratch in CUDA and C++. Implements custom CUDA kernels for tiled matrix multiplication, LayerNorm,…☆44May 17, 2026Updated 5 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☆17Nov 27, 2025Updated 10 months ago
- Wordpress hosting with auto-scaling - Free Trial Offer • AdFully Managed hosting for WordPress and WooCommerce businesses that need reliable, auto-scalable performance. Cloudways SafeUpdates now available.
- Using Rainbow implementation in Chainer RL for Slime Volleyball Pixel Environment☆23Jun 8, 2020Updated 6 years ago
- 使用mnn-llm对GOT-OCR2.0进行推理☆13Oct 2, 2024Updated 2 years ago
- ☆183Sep 18, 2026Updated last month
- Flash Attention in raw Cuda C beating PyTorch☆39May 14, 2024Updated 2 years 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
- 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
- segmentation algorithm yolact use tensorrt deploy☆14May 7, 2022Updated 4 years ago
- ☆19Jan 4, 2024Updated 2 years ago
- A lightweight, single-header C++11 Jinja2 template engine for LLM chat templates.☆20Mar 4, 2026Updated 7 months 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.
- Quantized LLM training in pure CUDA/C++.☆262Aug 31, 2026Updated last month
- arkit demo☆11Aug 20, 2018Updated 8 years ago
- Inference Llama 2 in one file of pure Cuda☆17Aug 20, 2023Updated 3 years ago
- ☆19Jun 6, 2025Updated last year
- Kernel Library for Large Headdim Attention (64~1024, BF16/FP8/FP4), 1.5x~15x↑ vs PyTorch SDPA.☆340Updated this week
- Implementation of LogAvgExp for Pytorch☆37Apr 10, 2025Updated last year
- 一个轻量化的大模型推理框架☆23May 26, 2025Updated last year