Learn CUDA with PyTorch
☆369Jun 1, 2026Updated 2 months ago
Alternatives and similar repositories for learn-cuda
Users that are interested in learn-cuda are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Fastest kernels written from scratch☆603Updated this week
- ☆41Updated this week
- A Quirky Assortment of CuTe Kernels☆1,116Updated this week
- A Python DSL to write Nvidia PTX for Hopper and Blackwell in JAX and PyTorch☆374Jul 9, 2026Updated last month
- ☆198Aug 9, 2026Updated last week
- GPU virtual machines on DigitalOcean Gradient AI • AdGet to production fast with high-performance AMD and NVIDIA GPUs you can spin up in seconds. The definition of operational simplicity.
- Tile primitives for speedy kernels☆3,631Jul 13, 2026Updated last month
- From Minimal GEMM to Everything☆234Jul 9, 2026Updated last month
- https://github.com/gpu-mode/reference-kernels☆26Jul 4, 2026Updated last month
- A pure-Python implementation of the Nvidia CuTe layout algebra intended to be approachable and easy to learn.☆239Jun 29, 2026Updated last month
- CUTLASS and CuTe Examples☆138Nov 30, 2025Updated 8 months ago
- An Optimizer for Nvidia Compilers.☆128Updated this week
- ☆129May 10, 2026Updated 3 months ago
- CUDA kernels for linear attention variants, written in CuTe DSL and CUTLASS C++.☆538Updated this week
- ☆14Dec 22, 2024Updated last year
- 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.
- Official Problem Sets / Reference Kernels for the GPU MODE Leaderboard!☆296Jul 26, 2026Updated 3 weeks ago
- Github mirror of trition-lang/triton repo.☆188Updated this week
- Interactive version of the CuTe layout paper☆57Apr 14, 2026Updated 4 months ago
- Write a fast kernel and see how you compare against the best humans and AI on gpumode.com☆110Jul 29, 2026Updated 2 weeks ago
- This repository contains companion software for the Colfax Research paper "Categorical Foundations for CuTe Layouts".☆142Updated this week
- ☆119May 31, 2025Updated last year
- ☆51Updated this week
- ☆19Mar 29, 2026Updated 4 months ago
- Accelerating MoE with IO and Tile-aware Optimizations☆743Updated this week
- GPU virtual machines on DigitalOcean Gradient AI • AdGet to production fast with high-performance AMD and NVIDIA GPUs you can spin up in seconds. The definition of operational simplicity.
- mKernel: fast multi-node, multi-GPU fused kernels☆266Updated this week
- High-performance GPU kernels written in TIRx.☆88Updated this week
- Explore training for quantized models☆28Jul 12, 2025Updated last year
- GPU programming related news and material links☆2,277Jun 15, 2026Updated 2 months ago
- A Python-embedded DSL that makes it easy to write fast, scalable ML kernels with minimal boilerplate.☆922Updated this week
- Some funny cute/cuteDSL code snippets☆33Mar 2, 2026Updated 5 months ago
- ☆189May 11, 2026Updated 3 months ago
- A curated list of resources for learning and exploring Triton, OpenAI's programming language for writing efficient GPU code.☆495Mar 10, 2025Updated last year
- Row-wise block scaling for fp8 quantization matrix multiplication. Solution to GPU mode AMD challenge.☆19Feb 9, 2026Updated 6 months ago
- GPU virtual machines on DigitalOcean Gradient AI • AdGet to production fast with high-performance AMD and NVIDIA GPUs you can spin up in seconds. The definition of operational simplicity.
- Fast and Memory-Efficient Exact Attention for Large Headdim, 1.5x~6x speedup over PyTorch SDPA.☆322Updated this week
- Fast CUDA matrix multiplication from scratch☆1,283Sep 2, 2025Updated 11 months ago
- FlashInfer: Kernel Library for LLM Serving☆6,179Updated this week
- High Performance LLM Inference Operator Library☆1,118Aug 6, 2026Updated last week
- Cute layout visualization☆45Jan 18, 2026Updated 7 months ago
- Flash Attention from Scratch on CUDA Ampere☆191Sep 1, 2025Updated 11 months ago
- Distributed Compiler based on Triton for Parallel Systems☆1,518Updated this week