Warp

英伟达的 Python 框架，把 Python 函数编译成 GPU 代码来写仿真

Warp 是英伟达开源的 Python 框架，用来写高性能的仿真和图形计算代码。开发者用 Python 写函数并加上 @wp.kernel 标记，Warp 在运行时把它即时编译成 CUDA 或 CPU 代码并行执行，不必直接写 C++/CUDA。它还支持自动微分，写出来的仿真可以对输入求梯度，即可微仿真。Warp 内置了几何、碰撞、粒子等常用工具，并能和 PyTorch、JAX 交换数据。英伟达的 Newton 物理引擎和 MuJoCo Warp 都建立在它之上。

例子用几十行 Warp 代码写一个布料仿真，在 GPU 上同时推进上千个并行环境，再把梯度传回 PyTorch 优化抓取轨迹。

也叫

NVIDIA Warp

相关

Newton 物理引擎、MuJoCo Warp、可微仿真、CUDA、GPU 并行仿真、算子

来源

NVIDIA/warp (GitHub)

信息截至

2025