Craftax: Ein blitzschneller JAX-basierter Benchmark für Open-Ended Reinforcement Learning

Craftax ist eine vollständig in JAX geschriebene RL-Umgebung, die die Spielmechanik von Crafter neu implementiert und erheblich erweitert und sich von Roguelike-Spielen wie NetHack inspirieren lässt. Kompatibel mit der gymnax-Schnittstelle lässt sie sich einfach in bestehende JAX-basierte Frameworks wie PureJaxRL und JaxUED integrieren. Auf der ICML 2024 als Spotlight-Paper angenommen, bietet Craftax CPU- und GPU/TPU-Unterstützung sowie Tutorials und Basislinien. Installationsanweisungen und Anwendungsbeispiele werden bereitgestellt.