Craftax: JAX 기반 초고속 오픈엔드 강화학습 벤치마크

Craftax는 JAX로 완전히 작성된 강화학습 환경으로, Crafter의 게임 메커니즘을 재구현하고 NetHack과 같은 로그라이크 게임에서 영감을 얻어 크게 확장했습니다. gymnax 인터페이스와 호환되므로 PureJaxRL 및 JaxUED와 같은 기존 JAX 기반 프레임워크와 쉽게 통합할 수 있습니다. ICML 2024에서 스포트라이트 논문으로 채택되었으며, Craftax는 CPU와 GPU/TPU를 모두 지원하며, 튜토리얼과 기준선이 포함되어 있습니다. 설치 지침과 사용 예시가 제공됩니다.