Toni-SM/skrl

Modular Reinforcement Learning (RL) library (implemented in PyTorch, JAX, and NVIDIA Warp) with support for Gymnasium/Gym, NVIDIA Isaac Lab, MuJoCo Playground and other environments

何を解決するか

skrl は強化学習(RL)アルゴリズムを実装・使用するためのモジュール化され、透明性の高い方法を提供します。さまざまな環境インターフェース間でのエージェントのトレーニングを簡素化し、柔軟な設定と、1回の実行内で環境のサブセット(スコープ)を同時にトレーニングできるようにします。

動作方法

このライブラリは Python で実装されており、PyTorch、JAX、NVIDIA Warp などの複数のバックエンドをサポートしています。OpenAI Gym、Farama Gymnasium、PettingZoo、ManiSkill、NVIDIA Isaac Lab、MuJoCo Playground などの幅広い環境インターフェースと統合されています。

対象ユーザー

多様な環境を扱い、複数のディープラーニングフレームワークをサポートする、モジュール化され、読みやすく、シンプルな RL ライブラリを必要とする研究者や開発者向けです。

特徴

  • マルチバックエンド対応: PyTorch、JAX、NVIDIA Warp に対応。
  • Hugging Face 統合: モデルは Hugging Face で利用可能。
  • 広範な環境互換性: Gymnasium、PettingZoo、ManiSkill、および Isaac Lab や MuJoCo Playground などの専用シミュレータをサポート。
  • 柔軟なトレーニング: 1回の実行内でスコープ(環境のサブセット)ごとにエージェントを同時にトレーニング可能。

関連

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