AWSがAmazon SageMaker AIでNVIDIA Isaac Labを使用したロボット強化学習の訓練をスケーリング

AWS は Amazon SageMaker AI 上で NVIDIA Isaac Lab を使用してロボット制御ポリシーの訓練を行うソリューションを提供。Unitree H1 ヒューマノイドロボットが荒れた地形での速度制御を学習する Isaac-Velocity-Rough-H1-v0 タスクで実証。

SageMaker HyperPod（分散訓練向けの管理インフラ）と SageMaker Training Jobs（オンデマンド短期実行）の 2 つのコンピュート選択肢を提供。HyperPod は複数ノード障害時に自動的にノードを再起動または交換し、チェックポイントから訓練を再開する。

単一の Docker イメージと共有設定ファイルから生成されたスクリプトにより、同じ訓練コードが両方のバックエンド上で動作。Amazon SageMaker managed MLflow を使用して実験メトリクスを追跡可能。