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보다 느립니다
• 비교적 새로운 라이브러리로, 기존 대안보다 커뮤니티가 작습니다
• 기구학 문제에 한정 — 동역학이나 힘 제어는 지원하지 않습니다
• 문서가 아직 보강 중입니다