Raphael Rueda ретвитнул

Launching pyptx — a Python DSL for writing NVIDIA PTX kernels.
One PTX instruction = one Python call. Write pure PTX in Python.
Direct Hopper + Blackwell support: wgmma, TMA, tcgen05, mbarriers. JAX + PyTorch integration.
Includes GEMM, grouped GEMM, RMSNorm, SwiGLU, and a PTX→Python transpiler
pip install pyptx[torch]
pip install pyptx[jax]
github.com/patrick-toulme…
English
























