Nvidia Warp Python Packages

warp-lang

A Python framework for GPU-accelerated simulation, robotics, and machine learning.

mujoco-warp

GPU-optimized version of the MuJoCo physics simulator, designed for NVIDIA hardware.

nvidia-physicsnemo

Open-source deep-learning framework for building, training, and fine-tuning deep learning models using state-of-the-art Physics-ML methods

nvidia-modulus

Open-source deep-learning framework for building, training, and fine-tuning deep learning models using state-of-the-art Physics-ML methods

xlb

XLB: Accelerated Lattice Boltzmann (XLB) for Physics-based ML

warp-nn

CUDA Graphable Neural Networks for NVIDIA Warp

newton-physics

An open-source, GPU-accelerated physics simulation engine built upon NVIDIA Warp, specifically targeting roboticists and simulation researchers.

rewarped

A platform for reinforcement learning in differentiable multiphysics simulation