Craftax : Un benchmark ultrarapide basé sur JAX pour l'apprentissage par renforcement ouvert

Craftax est un environnement RL entièrement écrit en JAX, qui réimplémente et étend considérablement la mécanique de jeu de Crafter, s'inspirant de jeux roguelike comme NetHack. Compatible avec l'interface gymnax, il s'intègre facilement aux frameworks JAX tels que PureJaxRL et JaxUED. Accepté comme article vedette à ICML 2024, Craftax offre une prise en charge de la CPU et du GPU/TPU, ainsi que des tutoriels et des lignes de base. Des instructions d'installation et des exemples d'utilisation sont fournis.