Publicado em 25 de abril de 2026, pyptx é apresentado como uma DSL em Python para escrever kernels NVIDIA PTX, com suporte a Hopper e Blackwell e integração com JAX e PyTorch.
Publicado em 25 de abril de 2026, pyptx é uma DSL em Python para escrever kernels NVIDIA PTX. A descrição informa suporte às arquiteturas Hopper e Blackwell e integração com JAX e PyTorch.
Entre os exemplos citados estão GEMM, grouped GEMM, RMSNorm e SwiGLU. Para avaliar a ferramenta, consulte a documentação e os exemplos originais e verifique, no seu ambiente, a compatibilidade com a arquitetura e os frameworks pretendidos; a descrição não apresenta resultados de desempenho.