NVIDIA Warp：用于高性能GPU模拟和图形的Python框架 (github.com)

Warp是一个Python框架，用于编写高性能模拟和图形代码。Warp采用常规的Python函数，并将它们JIT编译成可以在CPU或GPU上运行的高效内核代码。Warp专为空间计算而设计，并配有一组丰富的原语，可以轻松编写用于物理模拟、感知、机器人和几何处理的程序。此外，Warp内核是可微分的，可以用作PyTorch和JAX等框架的机器学习管道的一部分。