A Python DSL to write Nvidia PTX for Hopper and Blackwell in JAX and PyTorch
☆385Jul 9, 2026Updated 3 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.☆255Jun 29, 2026Updated 4 months ago
- A Quirky Assortment of CuTe Kernels☆1,185Updated this week
- A Python-embedded DSL that makes it easy to write fast, scalable ML kernels with minimal boilerplate.☆977Updated this week
- Region-level profiling for CUDA kernels with trace, NVBit, CUPTI, NSys, and an interactive Explorer.☆133Apr 17, 2026Updated 6 months ago
- FLA but cuTile☆27Apr 17, 2026Updated 6 months ago
- Deploy to Railway using AI coding agents - Free Credits Offer • AdUse Claude Code, Codex, OpenCode, and more. Autonomous software development now has the infrastructure to match with Railway.
- Unified tracing profiler and visualizer, one timeline from CPU to GPU/HES to in-kernel zones☆150Sep 21, 2026Updated last month
- Helpful kernel tutorials, examples and SKILLs for tile-based GPU programming☆830Updated this week
- Nsight Python is a Python kernel profiling interface based on NVIDIA Nsight Tools☆301Updated this week
- Reference implementation and examples of the CuTe Layout representation and algebra.☆382Oct 22, 2026Updated last week
- A dynamic binary instrumentation tool for tracing and analyzing GPU kernel instructions.☆84Updated this week
- Tile primitives for speedy kernels☆3,786Sep 12, 2026Updated last month
- Reverse engineering NVIDIA SASS instruction dictionary, kernel audits and pattern recognition across GPU architectures.☆350May 18, 2026Updated 5 months ago
- PTX ISA 9.1 documentation converted to searchable markdown. Includes Claude Code skill for CUDA development.☆231Dec 24, 2025Updated 10 months ago
- An agent harness that compiles a model into one provably-correct, self-retargeting CUDA megakernel and self-tunes it past cuBLAS at batch…☆157Sep 18, 2026Updated last month
- Deploy to Railway using AI coding agents - Free Credits Offer • AdUse Claude Code, Codex, OpenCode, and more. Autonomous software development now has the infrastructure to match with Railway.
- A collection of megakernels for TPU☆173Oct 15, 2026Updated 2 weeks ago
- CUDA kernels for linear attention variants, written in CuTe DSL and CUTLASS C++.☆560Updated this week
- This repository contains companion software for the Colfax Research paper "Categorical Foundations for CuTe Layouts".☆147Aug 15, 2026Updated 2 months ago
- Learn CUDA with PyTorch☆522Oct 3, 2026Updated 3 weeks ago
- ☆58Sep 27, 2026Updated last month
- cuTile is a programming model for writing parallel kernels for NVIDIA GPUs☆2,163Updated this week
- Github mirror of trition-lang/triton repo.☆205Updated this week
- Tilus is a tile-level kernel programming language with explicit control over shared memory and registers.☆497Oct 23, 2026Updated last week
- ☆40Dec 14, 2025Updated 10 months ago
- Managed Kubernetes at scale on DigitalOcean • AdDigitalOcean Kubernetes includes the control plane, bandwidth allowance, container registry, automatic updates, and more for free.
- CUDA Tile IR is an MLIR-based intermediate representation and compiler infrastructure for CUDA kernel optimization, focusing on tile-base…☆1,047Oct 16, 2026Updated 2 weeks ago
- ☆65Apr 26, 2025Updated last year
- Fused MoE kernel☆23Oct 23, 2026Updated last week
- Mirage Persistent Kernel: Compiling LLMs into a MegaKernel☆2,586Updated this week
- ☆68Feb 24, 2026Updated 8 months ago
- a simple API to use CUPTI☆10Aug 19, 2025Updated last year
- Fastest kernels written from scratch☆629Aug 15, 2026Updated 2 months ago
- Interactive version of the CuTe layout paper☆56Apr 14, 2026Updated 6 months ago
- Accelerating MoE with IO and Tile-aware Optimizations☆785Aug 29, 2026Updated 2 months ago
- Managed Kubernetes at scale on DigitalOcean • AdDigitalOcean Kubernetes includes the control plane, bandwidth allowance, container registry, automatic updates, and more for free.
- An Optimizer for Nvidia Compilers.☆144Oct 21, 2026Updated last week
- TensorIR is a lightweight NVIDIA-owned MLIR compiler frontend for expressing tensor computations and lowering them to NVIDIA CUDA Tile IR…☆141Updated this week
- Autoresearch for GPU kernels. Give it any PyTorch model, go to sleep, wake up to optimized Triton kernels.☆1,614Oct 23, 2026Updated last week
- FlyDSL is the Python front‑end of the project: a Flexible Layout Python DSL for expressing tiling, partitioning, data movement, and kerne…☆305Updated this week
- mKernel: fast multi-node, multi-GPU fused kernels☆306Oct 17, 2026Updated last week
- An experimental communicating attention kernel based on DeepEP.☆34Jul 29, 2025Updated last year
- CODA: Rewriting Transformer Blocks as GEMM-Epilogue Programs☆258Updated this week