Wrap：来自NVIDIA开发的基于Python的仿真框架

1天前浏览36

Warp是来自NVIDIA的工程师开发的用于编写高性能模拟和图形代码的Python框架。Warp 采用常规的Python函数，并对其进行JIT编译为可以在CPU或GPU上运行的高效内核代码。

Warp 专为空间计算而设计，拥有丰富的基元集，可轻松编写用于物理模拟、感知、机器人和几何处理的程序。此外，Warp 内核是可微分的，可与 PyTorch、JAX和Paddle等框架一起用作机器学习管道的一部分。

建议使用Python版本3.9或更高版本。Warp 可以在Windows、Linux和macOS上的 x86-64和ARMv8 CPU上运行。GPU支持需要支持CUDA的NVIDIA GPU和驱动程序（最低GeForce GTX 9xx）。

pip的安装方式如下：

pip install warp-lang

这里有专门来自NVIDIA的Warp工程总监在2024年3月份对该项目和产品的介绍分享（文末有PDF附件下载）：Warp: Advancing Simulation AI with Differentiable GPU Computing in Python。

分享中介绍了Warp中用于3D数据生成、计算机辅助工程和机器人技术的最新功能，并展示Warp如何无缝连接到PyTorch和JAX等机器学习框架。同时通过深入的示例来说明这些概念，包括飞行器的轨迹优化、有限元分析和大规模计算流体动力学。最后，介绍了Warp路线图，以及即将推出的功能将如何使用户能够利用Tensor Core的强大功能来加速神经网络推理和训练。