Craftax：基于JAX的超高速开放式强化学习基准

Craftax是一个完全用JAX编写的强化学习环境，它重新实现并显著扩展了Crafter游戏机制，并从NetHack等Roguelike游戏中汲取灵感。Craftax兼容gymnax接口，易于与PureJaxRL和JaxUED等现有基于JAX的框架集成。它已在ICML 2024上作为重点论文被接收。Craftax提供CPU和GPU/TPU版本，并包含教程和基准测试代码。