NVIDIA Warp et MJWarp pour la simulation robotique

OOndine

Hugging Face présente le passage d’un bras robotique SO-101 d’un workflow MuJoCo classique à MJWarp, qui s’appuie sur NVIDIA Warp pour exécuter la simulation sur processeur graphique. L’article détaille l’association de MuJoCo avec MJWarp, puis la préparation de cette configuration et sa mise à l’échelle pour faire tourner plusieurs simulations robotiques en parallèle. Dans l’exemple, jusqu’à 2 048 environnements MJWarp sont mis en place simultanément, ce qui illustre l’échelle de simulation présentée. NVIDIA Warp est un framework Python qui permet d’écrire des noyaux de calcul parallèles, autrement dit des opérations conçues pour être exécutées en parallèle, et de les compiler pour le processeur central ou pour CUDA. L’article souligne aussi la gestion explicite des tâches parallèles et des tableaux sur le périphérique de calcul, des éléments mobilisés pour préparer cette exécution parallèle. Warp prend en charge la différentiation et, depuis la version 1.15, des modes d’exécution déterministes, mais ces propriétés ne garantissent pas à elles seules la différentiabilité ou le déterminisme d’une simulation MJWarp complète. Le travail présenté porte sur la simulation et sa mise à l’échelle, sans entraîner de politique robotique : il décrit donc la préparation de simulations parallèles, pas l’apprentissage du comportement du bras.