Nvidia Warp (a Python framework for writing high-performance code)

Warp is a Python framework for writing high-performance simulation and graphics code. Warp takes
regular Python functions and JIT compiles them to efficient kernel code that can run on the CPU or GPU. Read more

Similar