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

warp-nn

CUDA Graphable Neural Networks for NVIDIA Warp

nvidia-modulus

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

warp-simdata

A Python library built on NVIDIA Warp for GPU-accelerated simulation data processing, with reusable operators that work across many types of simulation data.

physicsnemo

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

newton-physics

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

rewarped

Reinforcement learning in differentiable multiphysics simulation with NVIDIA Warp.