PyRoki

Boîte à outils Python modulaire pour l'optimisation cinématique sur CPU, GPU et TPU via JAX

Licence

MIT

Aperçu

Une boîte à outils Python modulaire, extensible et multiplateforme pour l'optimisation cinématique. Elle unifie la cinématique inverse, l'optimisation de trajectoire et le reciblage de mouvement avec des coûts et des contraintes composables. Elle fonctionne sur CPU, GPU et TPU via JAX. Elle prend en charge le chargement d'URDF avec différentiation automatique, la détection de collisions et plusieurs méthodes de résolution, dont Levenberg-Marquardt et le lagrangien augmenté. Développée à UC Berkeley.

PyRoki est une boîte à outils Python modulaire qui fournit un cadre unifié pour résoudre divers problèmes d'optimisation cinématique en robotique, notamment la cinématique inverse, l'optimisation de trajectoire et le reciblage de mouvement entre différentes structures cinématiques. Développée à l'UC Berkeley, la boîte à outils utilise JAX comme moteur de calcul, ce qui permet une exécution transparente sur CPU, GPU et TPU tout en fournissant la différentiation automatique pour l'optimisation par gradient. PyRoki prend en charge le chargement de fichiers URDF standard pour la description des robots, intègre des capacités de détection de collisions et implémente l'algorithme de Levenberg-Marquardt pour une convergence robuste. En unifiant plusieurs types de problèmes cinématiques dans un même cadre, PyRoki évite aux chercheurs de maintenir des outils spécialisés distincts pour différentes tâches de planification de mouvement, ce qui simplifie les flux de travail pour des applications allant de la téléopération au reciblage sur humanoïdes.

Pourquoi c'est important

PyRoKi offre une boîte à outils moderne et épurée pour la cinématique robotique, qui fonctionne sur toutes les plateformes. Son approche d'optimisation composable unifie l'IK, la planification de trajectoires et le reciblage de mouvement — des tâches qui nécessitaient auparavant des bibliothèques distinctes et incompatibles.

Le problème résolu

Les outils de cinématique robotique sont généralement propres à une plateforme et ne résolvent que des problèmes étroits. PyRoKi fournit un framework d'optimisation unifié dans lequel l'IK, la planification de trajectoires et le reciblage s'expriment tous comme des problèmes d'optimisation composables.

Points clés

• Boîte à outils Python modulaire pour l'optimisation cinématique
• Coûts et contraintes composables pour une formulation flexible des problèmes
• Unifie l'IK, l'optimisation de trajectoires et le reciblage de mouvement
• Multiplateforme, sans dépendances lourdes
• Architecture extensible pour des problèmes cinématiques personnalisés

Pour qui

• Ingénieurs roboticiens qui résolvent des problèmes de cinématique
• Chercheurs en reciblage de mouvement et en téléopération
• Développeurs ayant besoin d'une IK légère sans dépendance à ROS
• Équipes qui développent des logiciels robotiques multiplateformes

Limites

• L'approche par optimisation est plus lente que l'IK analytique pour les robots simples
• Bibliothèque plus récente, avec une communauté plus restreinte que les alternatives établies
• Limité aux problèmes cinématiques — pas de dynamique ni de contrôle en effort
• Documentation encore en cours d'enrichissement