Craftax：JAXベースの超高速オープンエンド強化学習ベンチマーク
2025-02-11
CraftaxはJAXで完全に記述された強化学習環境であり、Crafterのゲームメカニクスを再実装し、NetHackなどのローグライクゲームからインスピレーションを得て大幅に拡張しています。gymnaxインターフェースに準拠しており、PureJaxRLやJaxUEDなどの既存のJAXベースのフレームワークと容易に統合できます。ICML 2024でスポットライト論文として採択されました。CraftaxはCPUとGPU/TPUの両方をサポートしており、チュートリアルとベースラインが含まれています。インストール手順と使用例が提供されています。
開発
ゲーム環境