A Python DSL to write Nvidia PTX for Hopper and Blackwell in JAX and PyTorch - View it on GitHub