A Python DSL to write Nvidia PTX for Hopper and Blackwell in JAX and PyTorch
☆382Jul 9, 2026Updated 2 months ago
Alternatives and similar repositories for pyptx
Users that are interested in pyptx 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 pure-Python implementation of the Nvidia CuTe layout algebra intended to be approachable and easy to learn.☆243Jun 29, 2026Updated 2 months ago
- A Quirky Assortment of CuTe Kernels☆1,147Updated this week
- A Python-embedded DSL that makes it easy to write fast, scalable ML kernels with minimal boilerplate.☆948Updated this week
- Unified tracing profiler and visualizer, one timeline from CPU to GPU/HES to in-kernel zones☆140Sep 12, 2026Updated last week
- FLA but cuTile☆27Apr 17, 2026Updated 5 months 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.
- Region-level profiling for CUDA kernels with trace, NVBit, CUPTI, NSys, and an interactive Explorer.☆130Apr 17, 2026Updated 5 months ago
- A dynamic binary instrumentation tool for tracing and analyzing GPU kernel instructions.☆79Updated this week
- Nsight Python is a Python kernel profiling interface based on NVIDIA Nsight Tools☆295Updated this week
- PTX ISA 9.1 documentation converted to searchable markdown. Includes Claude Code skill for CUDA development.☆228Dec 24, 2025Updated 8 months ago
- CUDA kernels for linear attention variants, written in CuTe DSL and CUTLASS C++.☆548Sep 8, 2026Updated last week
- Tile primitives for speedy kernels☆3,716Sep 12, 2026Updated last week
- Reverse engineering NVIDIA SASS instruction dictionary, kernel audits and pattern recognition across GPU architectures.☆340May 18, 2026Updated 4 months ago
- FlashKDA: high-performance Kimi Delta Attention kernels☆1,254Sep 1, 2026Updated 2 weeks ago
- This repository contains companion software for the Colfax Research paper "Categorical Foundations for CuTe Layouts".☆145Aug 15, 2026Updated last month
- 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.
- Helpful kernel tutorials, examples and SKILLs for tile-based GPU programming☆810Updated this week
- Learn CUDA with PyTorch☆483Updated this week
- ☆39Dec 14, 2025Updated 9 months ago
- An Optimizer for Nvidia Compilers.☆136Updated this week
- An agent harness that compiles a model into one provably-correct, self-retargeting CUDA megakernel and self-tunes it past cuBLAS at batch…☆142Updated this week
- cuTile is a programming model for writing parallel kernels for NVIDIA GPUs☆2,150Updated this week
- CUDA Tile IR is an MLIR-based intermediate representation and compiler infrastructure for CUDA kernel optimization, focusing on tile-base…☆1,032Sep 10, 2026Updated last week
- Github mirror of trition-lang/triton repo.☆196Updated this week
- a simple API to use CUPTI☆10Aug 19, 2025Updated 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.
- Tilus is a tile-level kernel programming language with explicit control over shared memory and registers.☆492Updated this week
- Interactive version of the CuTe layout paper☆57Apr 14, 2026Updated 5 months ago
- Accelerating MoE with IO and Tile-aware Optimizations☆769Aug 29, 2026Updated 3 weeks ago
- FlyDSL is the Python front‑end of the project: a Flexible Layout Python DSL for expressing tiling, partitioning, data movement, and kerne…☆282Updated this week
- mKernel: fast multi-node, multi-GPU fused kernels☆277Updated this week
- ☆51Aug 17, 2026Updated last month
- TileFusion is an experimental C++ macro kernel template library that elevates the abstraction level in CUDA C for tile processing.☆118Aug 4, 2026Updated last month
- An experimental communicating attention kernel based on DeepEP.☆34Jul 29, 2025Updated last year
- Fastest kernels written from scratch☆623Aug 15, 2026Updated last month
- 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.
- Benchmark tests supporting the TiledCUDA library.☆19Nov 19, 2024Updated last year
- Mirage Persistent Kernel: Compiling LLMs into a MegaKernel☆2,504Sep 12, 2026Updated last week
- ☆457Aug 26, 2026Updated 3 weeks ago
- Autoresearch for GPU kernels. Give it any PyTorch model, go to sleep, wake up to optimized Triton kernels.☆1,561Mar 19, 2026Updated 6 months ago
- Learning TileLang with 10 puzzles!☆380Aug 11, 2026Updated last month
- Pure C / AVX-512 port of Craftax-Classic. 47.8M SPS on a Ryzen 9 9950X3D -- 3.2x an RTX Pro 6000 Blackwell on the same env.☆24Jul 14, 2026Updated 2 months ago
- Domain-specific language designed to streamline the development of high-performance GPU/CPU/Accelerators kernels☆7,433Updated this week