warp

NVIDIA Warp 是一个面向 GPU 加速计算的 Python 框架，用于高效执行物理仿真、机器人建模与可微分机器学习任务。它通过 JIT 编译将标准 Python 函数转换为 CUDA 内核，支持 CPU/GPU 后端统一调度；内置可微分物理原语（如刚体动力学、软体、碰撞检测）和几何处理算子，并原生兼容 PyTorch、JAX 等主流 ML 框架。适用于需要高吞吐仿真与梯度回传结合的研究与工程场景，如机器人运动规划、数字孪生仿真、神经渲染中的物理约束建模等。