“Warp is a Python framework for high-performance simulation and graphics code. It JIT compiles regular Python functions for efficient kernel code running on the CPU or GPU, perfect for physics simulation, robotics, and more. Warp kernels are differentiable, useful for machine learning with PyTorch and JAX. The Github repository offers examples, and users can install Warp from PyPI or source. Some examples require a CUDA-capable device, and USD files can be viewed in Omniverse, UsdView, and Blender. Extensions for Warp in Omniverse are available. Warp has been used in NVIDIA research projects. Visit GitHub for support and contribute to this project under the NVIDIA Software License.”
https://nvidia.github.io/warp/