warp

NVIDIA Warp is a Python framework designed for accelerated simulation and spatial computing, enabling high-performance physics simulations and data generation. It supports differentiable programming, making it suitable for integration into machine learning pipelines, particularly in the context of molecular simulations.