Step by step implementation of a fast softmax kernel in CUDA
☆63Jan 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:
- ☆93Nov 11, 2025Updated 4 months ago
- 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
- learn TensorRT from scratch🥰☆17Sep 29, 2024Updated last year
- Row-wise block scaling for fp8 quantization matrix multiplication. Solution to GPU mode AMD challenge.☆18Feb 9, 2026Updated last month
- Managed Database hosting by DigitalOcean • AdPostgreSQL, MySQL, MongoDB, Kafka, Valkey, and OpenSearch available. Automatically scale up storage and focus on building your apps.
- A collection of GPU experiments and benchmarks for my personal understanding and research.☆26Mar 18, 2026Updated last week
- Standalone commandline CLI tool for compiling Triton kernels☆20Sep 13, 2024Updated last year
- ☆19Mar 12, 2025Updated last year
- Official Problem Sets / Reference Kernels for the GPU MODE Leaderboard!☆226Mar 18, 2026Updated last 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 9 months ago
- Super fast FP32 matrix multiplication on RDNA3☆87Mar 30, 2025Updated 11 months ago
- ☆53Feb 24, 2026Updated last month
- Flash Attention in raw Cuda C beating PyTorch☆38May 14, 2024Updated last year
- Official repository for the paper Local Linear Attention: An Optimal Interpolation of Linear and Softmax Attention For Test-Time Regressi…☆23Oct 1, 2025Updated 5 months ago
- Bare Metal GPUs on DigitalOcean Gradient AI • AdPurpose-built for serious AI teams training foundational models, running large-scale inference, and pushing the boundaries of what's possible.
- ☆16Feb 24, 2026Updated last month
- General Matrix Multiplication using NVIDIA Tensor Cores☆28Jan 25, 2025Updated last year
- ☆13Jul 2, 2025Updated 8 months ago
- Flash Attention in ~100 lines of CUDA (forward pass only)☆10Jun 10, 2024Updated last year
- ☆13Nov 3, 2025Updated 4 months ago
- [CVPR 2025 Highlight] FIMA-Q: Post-Training Quantization for Vision Transformers by Fisher Information Matrix Approximation☆25Jun 16, 2025Updated 9 months ago
- Fast CUDA matrix multiplication from scratch☆1,100Sep 2, 2025Updated 6 months ago
- 搜藏的希望的代码片段☆13Jun 6, 2023Updated 2 years ago
- HunyuanDiT with TensorRT and libtorch☆18May 22, 2024Updated last year
- 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.
- Cute layout visualization☆33Jan 18, 2026Updated 2 months ago
- TensorRT-in-Action 是一个 GitHub 代码库，提供了使用 TensorRT 的代码示例，并有对应 Jupyter Notebook。☆15Jun 1, 2023Updated 2 years ago
- 使用mnn-llm对GOT-OCR2.0进行推理☆14Oct 2, 2024Updated last year
- ☆15Feb 23, 2025Updated last year
- segmentation algorithm yolact use tensorrt deploy☆14May 7, 2022Updated 3 years ago
- 📚 A curated list of awesome matrix-matrix multiplication (A * B = C) frameworks, libraries and software☆64Feb 23, 2025Updated last year
- 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
- Repository to host ROCm Developer Hub Notebook Tutorials☆58Updated this week
- 训练营讲义☆20Jan 21, 2025Updated last year
- 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.
- Inference Llama 2 in one file of pure Cuda☆17Aug 20, 2023Updated 2 years ago
- ☆261Jul 11, 2024Updated last year
- A novel non-pretrained deep supervision network for polyp segmentation☆11Jun 5, 2024Updated last year
- THC: Accelerating Distributed Deep Learning Using Tensor Homomorphic Compression☆19Jul 30, 2024Updated last year
- Implement Flash Attention using Cute.☆103Dec 17, 2024Updated last year
- 一个轻量化的大模型推理框架☆22May 26, 2025Updated 9 months ago
- ☆22Apr 4, 2025Updated 11 months ago