Craftax：JAXベースの超高速オープンエンド強化学習ベンチマーク

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