Craftax: Um Benchmark de RL de Alta Velocidade em JAX para Aprendizado por Reforço Aberto

Craftax é um ambiente de RL totalmente escrito em JAX, reimplementando e expandindo significativamente a mecânica do jogo Crafter, inspirado em jogos roguelike como NetHack. Compatível com a interface gymnax, ele se integra facilmente a frameworks JAX como PureJaxRL e JaxUED. Aceito como artigo destacado na ICML 2024, o Craftax oferece suporte para CPU e GPU/TPU, além de tutoriais e linhas de base. Instruções de instalação e exemplos de uso são fornecidos.