PyRoki

JAX により CPU・GPU・TPU で動作する、運動学最適化のためのモジュール型 Python ツールキット

ライセンス

MIT

概要

運動学最適化のための、モジュラーで拡張可能なクロスプラットフォームのPythonツールキット。組み合わせ可能なコストと制約によって、逆運動学、軌道最適化、モーションリターゲティングを統合します。JAXによりCPU、GPU、TPUで動作します。自動微分に対応したURDFの読み込み、衝突検出、Levenberg-MarquardtやAugmented Lagrangianを含む複数のソルバー手法をサポートしています。UC Berkeleyで開発されました。

PyRokiは、ロボティクスにおけるさまざまな運動学的最適化問題を解くための統一フレームワークを提供するモジュール式Pythonツールキットです。対象には逆運動学、軌道最適化、異なる運動学構造間のモーションリターゲティングが含まれます。UC Berkeleyで開発されたこのツールキットは、計算バックエンドとしてJAXを活用しており、CPU、GPU、TPUのハードウェア上でシームレスに実行できるうえ、勾配ベースの最適化のための自動微分を提供します。PyRokiはロボット記述用の標準的なURDFファイルの読み込み、組み込みの衝突検出機能に対応し、堅牢な収束のためにLevenberg-Marquardtアルゴリズムを実装しています。複数の種類の運動学問題を単一のフレームワークに統合することで、研究者が異なるモーションプランニングタスクごとに専用ツールを別々に保守する必要をなくし、遠隔操作からヒューマノイドのリターゲティングまで幅広い用途のワークフローを効率化します。

重要な理由

PyRoKi は、プラットフォームを問わず動作する、すっきりとしたモダンなロボット運動学ツールキットです。組み合わせ可能な最適化アプローチにより、IK、軌道計画、モーションリターゲティングを統合します。これらは従来、別々で互換性のないライブラリを必要としていたタスクです。

解決する課題

ロボット運動学のツールは通常、特定のプラットフォームに依存し、限られた問題しか解けません。PyRoKi は統一された最適化フレームワークを提供し、IK、軌道計画、リターゲティングをすべて組み合わせ可能な最適化問題として表現します。

主なポイント

• 運動学最適化のためのモジュール型 Python ツールキット
• 組み合わせ可能なコストと制約による柔軟な問題定式化
• IK、軌道最適化、モーションリターゲティングを統合
• 重い依存関係のないクロスプラットフォーム設計
• カスタムの運動学問題に対応できる拡張可能なアーキテクチャ

対象

• 運動学の問題に取り組むロボティクスエンジニア
• モーションリターゲティングや遠隔操作の研究者
• ROS に依存しない軽量な IK を必要とする開発者
• クロスプラットフォームのロボットソフトウェアを開発するチーム

制限事項

• 単純なロボットでは、最適化ベースの手法は解析的 IK より遅い
• 比較的新しいライブラリで、定評ある代替手段よりコミュニティが小さい
• 運動学の問題に限定——動力学や力制御は扱わない
• ドキュメントはまだ充実の途上