warp

NVIDIA Warp 是一个用于GPU加速的仿真、机器人技术和机器学习的Python框架。它能够将普通的Python函数即时编译为可在CPU或GPU上运行的高效内核代码，支持物理模拟、几何处理等多种原语，并且这些内核是可微分的，可以与PyTorch、JAX等机器学习框架结合使用。特别适合需要高性能计算支持的研究者和开发者在进行复杂物理现象模拟、机器人控制算法开发以及深度学习模型训练时采用。