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.☆253Jun 29, 2026Updated 3 months ago
- A Quirky Assortment of CuTe Kernels☆1,167Sep 17, 2026Updated 3 weeks ago
- A Python-embedded DSL that makes it easy to write fast, scalable ML kernels with minimal boilerplate.☆960Updated this week
- Unified tracing profiler and visualizer, one timeline from CPU to GPU/HES to in-kernel zones☆148Sep 21, 2026Updated 2 weeks ago
- FLA but cuTile☆27Apr 17, 2026Updated 5 months ago
- Managed Database hosting by DigitalOcean • AdPostgreSQL, MySQL, MongoDB, Kafka, Valkey, and OpenSearch available. Automatically scale up storage and focus on building your apps.
- Region-level profiling for CUDA kernels with trace, NVBit, CUPTI, NSys, and an interactive Explorer.☆133Apr 17, 2026Updated 5 months ago
- A dynamic binary instrumentation tool for tracing and analyzing GPU kernel instructions.☆83Updated this week
- Nsight Python is a Python kernel profiling interface based on NVIDIA Nsight Tools☆297Updated this week
- PTX ISA 9.1 documentation converted to searchable markdown. Includes Claude Code skill for CUDA development.☆229Dec 24, 2025Updated 9 months ago
- CUDA kernels for linear attention variants, written in CuTe DSL and CUTLASS C++.☆555Updated this week
- Tile primitives for speedy kernels☆3,746Sep 12, 2026Updated 3 weeks ago
- Reverse engineering NVIDIA SASS instruction dictionary, kernel audits and pattern recognition across GPU architectures.☆342May 18, 2026Updated 4 months ago
- FlashKDA: high-performance Kimi Delta Attention kernels☆1,273Sep 1, 2026Updated last month
- This repository contains companion software for the Colfax Research paper "Categorical Foundations for CuTe Layouts".☆147Aug 15, 2026Updated last month
- Managed Kubernetes at scale on DigitalOcean • AdDigitalOcean Kubernetes includes the control plane, bandwidth allowance, container registry, automatic updates, and more for free.
- Helpful kernel tutorials, examples and SKILLs for tile-based GPU programming☆823Updated this week
- Learn CUDA with PyTorch☆498Oct 3, 2026Updated last week
- ☆40Dec 14, 2025Updated 9 months ago
- An Optimizer for Nvidia Compilers.☆137Sep 23, 2026Updated 2 weeks ago
- An agent harness that compiles a model into one provably-correct, self-retargeting CUDA megakernel and self-tunes it past cuBLAS at batch…☆148Sep 18, 2026Updated 3 weeks ago
- cuTile is a programming model for writing parallel kernels for NVIDIA GPUs☆2,155Updated this week
- CUDA Tile IR is an MLIR-based intermediate representation and compiler infrastructure for CUDA kernel optimization, focusing on tile-base…☆1,039Sep 28, 2026Updated last week
- Github mirror of trition-lang/triton repo.☆201Updated this week
- a simple API to use CUPTI☆10Aug 19, 2025Updated last year
- Managed Kubernetes at scale on DigitalOcean • AdDigitalOcean Kubernetes includes the control plane, bandwidth allowance, container registry, automatic updates, and more for free.
- Tilus is a tile-level kernel programming language with explicit control over shared memory and registers.☆495Sep 17, 2026Updated 3 weeks ago
- Interactive version of the CuTe layout paper☆57Apr 14, 2026Updated 5 months ago
- Accelerating MoE with IO and Tile-aware Optimizations☆778Aug 29, 2026Updated last month
- FlyDSL is the Python front‑end of the project: a Flexible Layout Python DSL for expressing tiling, partitioning, data movement, and kerne…☆291Updated this week
- mKernel: fast multi-node, multi-GPU fused kernels☆291Updated 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.☆119Aug 4, 2026Updated 2 months ago
- An experimental communicating attention kernel based on DeepEP.☆34Jul 29, 2025Updated last year
- Fastest kernels written from scratch☆628Aug 15, 2026Updated last month
- AI Agents on 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.
- Benchmark tests supporting the TiledCUDA library.☆19Nov 19, 2024Updated last year
- Mirage Persistent Kernel: Compiling LLMs into a MegaKernel☆2,547Updated this week
- ☆503Aug 26, 2026Updated last month
- Autoresearch for GPU kernels. Give it any PyTorch model, go to sleep, wake up to optimized Triton kernels.☆1,584Mar 19, 2026Updated 6 months ago
- 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
- Learning TileLang with 10 puzzles!☆386Aug 11, 2026Updated last month
- Domain-specific language designed to streamline the development of high-performance GPU/CPU/Accelerators kernels☆8,557Updated this week