NVIDIA/warp: A Python framework for high performance GPU simulation and graphics

Warp is a Python framework for high-performance GPU simulation and graphics. It supports physics simulation, perception, robotics, and geometry processing. Warp kernels are differentiable and can be used with frameworks like PyTorch and JAX. It can be installed from PyPI and requires Python 3.9 or newer.