NVIDIA presenta osmo: una herramienta de flujo de trabajo unificada para el desarrollo de robots con IA

NVIDIA ha presentado OSMO, un orquestador de flujo de trabajo de código abierto diseñado para simplificar el desarrollo de robots con IA. La herramienta, construida sobre Kubernetes, permite a los desarrolladores describir todo el flujo de trabajo, desde el entrenamiento hasta la simulación y las pruebas, en un único archivo YAML. Esto elimina la necesidad de código de infraestructura separado en diferentes etapas del desarrollo. OSMO aborda el proceso fragmentado del desarrollo de IA físico, donde el entrenamiento se realiza en GPUs de centros de datos, la simulación en hardware RTX de clase de estación de trabajo y el despliegue en dispositivos de borde como Jetson AGX Thor. Al tratar estas etapas como parte de un único plano de control, OSMO simplifica la orquestación de tareas en diferentes plataformas. El sistema enruta las tareas en función de los nombres de las plataformas, como gb200, rtx-pro-6000 o jetson-agx-thor, sin necesidad de referencias explícitas al clúster. La herramienta está licenciada bajo Apache-2.0, incluye gráficos Helm y contenedores en NGC, y ofrece un inicio rápido local para probar en una estación de trabajo. Admite grupos de tareas en serie y en paralelo, plantillas Jinja y programación basada en prioridades con compartición de recursos de GPU. NVIDIA destaca a OSMO como una solución a las complejidades de la gestión de múltiples pilas de herramientas en el desarrollo de IA físico.