dexmal/dexbotic

Dexbotic: Open-Source Vision-Language-Action Toolbox

何を解決するか

Dexboticは、視覚言語行動(VLA)モデルの統合開発ツールボックスを提供し、身体的な知能のための事前学習、微調整、推論、評価のプロセスを簡素化します。異なるVLAアルゴリズム用の複雑な環境を設定する手間を排除し、さまざまなロボットプラットフォーム間でトレーニングデータを標準化して扱う方法を提供します。

動作方法

PyTorchに基づいて構築されたこのツールボックスは、階層的な設定、ファクトリ登録、エントリディスパッチからなるモジュールアーキテクチャを採用しています。これにより、研究者はコアコードを再実装せずに、モデルの交換、設定の変更、新しいタスクの追加を実験スクリプトで行うことができます。ローカルの一般消費者向けGPUとクラウドトレーニングプラットフォームの両方をサポートしており、UR5、Franka、ALOHAなどの異なるハードウェア間で互換性を確保するための統一されたトレーニングデータフォーマットを含んでいます。

対象ユーザー

身体的AIの研究や開発に従事する研究者や開発者、特にロボット操作およびナビゲーション用のVLAモデルを開発している人向けです。

主な特徴

  • 広範なモデル対応: π0、CogACT、OFT、MemVLA、DM0などの主流ポリシーを統合しています。
  • 最適化されたパフォーマンス: 高性能な事前学習済み基盤モデルと、Tritonをバックエンドに持つリアルタイムバックエンドを備えており、推論速度を最大5倍に向上させます。
  • 柔軟なトレーニング: FSDP2による分散トレーニングと、効率的な微調整のためのLoRAをサポートしています。
  • ハードウェア非依存: 複数のロボットタイプ向けのデプロイスクリプトとデータフォーマットを提供し、最新のBlackwell GPUアーキテクチャもサポートしています。

関連

  • プロジェクト
  • プロジェクト
  • プロジェクト
  • プロジェクト