Stanford dévoile Frax : un calculateur robotique ultrarapide
Une équipe de chercheurs du laboratoire ASL (Autonomous Systems Lab) de l'université de Stanford a rendu publique une nouvelle bibliothèque open source, baptisée Frax, conçue pour accélérer considérablement les calculs de cinématique et de dynamique des robots. Disponible sur la plateforme GitHub sous le nom d'utilisateur StanfordASL, cette boîte à outils vise à répondre à un besoin croissant de rapidité et d'efficacité dans les algorithmes de contrôle et d'apprentissage robotique.
Le projet se distingue par son fondement sur JAX, une bibliothèque de calcul numérique développée par Google Research, connue pour sa capacité à combiner la compilation accélérée par GPU/TPU avec la différentiation automatique. En reposant sur JAX, Frax permet de calculer les équations du mouvement d'un robot — sa cinématique (positions, vitesses, accélérations) et sa dynamique (forces, couples) — de manière extrêmement rapide par rapport aux méthodes traditionnelles utilisant par exemple Python ou C++.
Des performances accrues pour l'apprentissage et le contrôle
L'intérêt principal de Frax réside dans la possibilité d'exécuter ces calculs intensifs en parallèle sur du matériel accélérateur (GPU, TPU). Cela ouvre des perspectives pour des applications telles que l'apprentissage par renforcement, où des milliers de simulations robotiques doivent être effectuées en parallèle pour entraîner des politiques de contrôle. La différentiation automatique, intégrée nativement dans JAX, simplifie également le calcul des gradients, essentiel pour les algorithmes d'optimisation et d'apprentissage.
Selon les informations disponibles sur le dépôt du projet, Frax est encore à un stade de développement, mais il suscite déjà un intérêt certain dans la communauté des roboticiens. La bibliothèque est publiée sous licence open source, permettant à tout chercheur ou développeur de l'utiliser, de la modifier et de contribuer à son amélioration.
Un exemple concret : le bras robotique Franka Emika Panda
Pour illustrer ses capacités, l'équipe de Stanford a fourni des exemples d'utilisation de Frax avec le modèle du bras robotique Franka Emika Panda, un manipulateur collaboratif courant en recherche. La bibliothèque permet non seulement de calculer sa cinématique directe et inverse, mais aussi sa dynamique complète, le tout en tirant parti des optimisations de JAX.
Implications et perspectives
Ce type d'outil pourrait avoir un impact significatif sur le développement de la robotique, en réduisant le temps de calcul nécessaire à la simulation et au contrôle de robots complexes. Les chercheurs espèrent que Frax facilitera la recherche sur des algorithmes plus avancés de planification de mouvement et de contrôle adaptatif. La mise à disposition de la bibliothèque en open source pourrait également favoriser son adoption rapide par la communauté scientifique et industrielle.
