NVIDIA WARP

Summary from the TLDR newsletter

NVIDIA Warp is a Python framework for writing high-performance simulation and graphics code. It takes regular Python functions and JIT compiles them to efficient kernel code that can run on the CPU or GPU. Warp comes with a rich set of primitives that make it easy to write programs for physics simulation, perception, robotics, and geometry processing. Its kernels can be used as part of machine learning pipelines with frameworks such as PyTorch and JAX.