dexmal/dexbotic

Dexbotic: Open-Source Vision-Language-Action Toolbox

해결하는 문제

Dexbotic은 시각-언어-행동(VLA) 모델을 위한 통합 개발 도구 상자를 제공하여, 신체적 지능을 위한 사전 훈련, 미세 조정, 추론 및 평가 과정을 간소화합니다. 다양한 VLA 알고리즘을 위한 복잡한 환경 설정의 번거로움을 제거하고, 다양한 로봇 플랫폼 간의 훈련 데이터를 표준화된 방식으로 처리할 수 있도록 합니다.

작동 방식

PyTorch 기반으로 구축된 이 도구 상자는 계층적 설정, 팩토리 등록, 엔트리 디스패치로 구성된 모듈식 아키텍처를 사용합니다. 연구자들은 핵심 코드를 재작성하지 않고도 실험 스크립트를 통해 모델을 교체하거나 설정을 수정하거나 새로운 작업을 추가할 수 있습니다. 로컬 소비자용 GPU와 클라우드 훈련 플랫폼을 모두 지원하며, UR5, Franka, ALOHA와 같은 다양한 하드웨어 간 호환성을 보장하는 통합 훈련 데이터 형식을 포함합니다.

대상 사용자

특히 로봇 조작 및 탐색을 위한 VLA 모델을 개발하는 신체적 AI 연구자 및 개발자에게 적합합니다.

주요 특징

  • 광범위한 모델 지원: π0, CogACT, OFT, MemVLA, DM0 등의 주요 정책을 통합합니다.
  • 최적화된 성능: 고성능 사전 훈련 기반 모델과 Triton 기반 실시간 백엔드를 포함하여 추론 속도를 최대 5배까지 향상시킵니다.
  • 유연한 훈련: 분산 훈련을 위한 FSDP2와 효율적인 미세 조정을 위한 LoRA를 지원합니다.
  • 하드웨어 독립성: 다양한 로봇 유형용 배포 스크립트와 데이터 형식을 제공하며 최신 Blackwell GPU 아키텍처도 지원합니다.

관련

  • 프로젝트
  • 프로젝트
  • 프로젝트
  • 프로젝트