Published on April 25, 2026, pyptx is described as a Python DSL for writing NVIDIA PTX kernels, with Hopper and Blackwell support and integration with JAX and PyTorch.
Published on April 25, 2026, pyptx is a Python DSL for writing NVIDIA PTX kernels. Its description states support for Hopper and Blackwell architectures and integration with JAX and PyTorch.
The examples cited include GEMM, grouped GEMM, RMSNorm, and SwiGLU. To assess the tool, consult the original documentation and examples, then verify compatibility in your intended environment; the description does not provide performance results.