Craftax: Un Benchmark de RL ultrarrápido basado en JAX para RL Abierto

Craftax es un entorno de RL escrito completamente en JAX, que reimplementa y extiende significativamente la mecánica de juego de Crafter, inspirándose en juegos roguelike como NetHack. Compatible con la interfaz gymnax, se integra fácilmente con frameworks JAX como PureJaxRL y JaxUED. Aceptado como artículo destacado en ICML 2024, Craftax ofrece soporte para CPU y GPU/TPU, junto con tutoriales y líneas de base. Se proporcionan instrucciones de instalación y ejemplos de uso.