Mahjax: Simulatore di Mahjong Accelerato su GPU per la Ricerca RL

Mahjax è un ambiente di Riichi Mahjong completamente vettorizzato implementato in JAX, progettato per consentire la parallelizzazione su larga scala delle rollout su GPU per la ricerca sull'apprendimento per rinforzo. Il Riichi Mahjong è un gioco multiplayer a informazione imperfetta, con stocasticità e spazi di stato ad alta dimensionalità, che presenta sfide simili a problemi decisionali del mondo reale. L'ambiente supporta l'apprendimento tabula rasa, il che significa che gli algoritmi possono imparare da zero senza fare affidamento su log di partite umane. Mahjax raggiunge throughput fino a 2 milioni di passi di gioco al secondo. Include anche uno strumento di visualizzazione di alta qualità per il debug e l'interazione con agenti addestrati. Il progetto mira a facilitare la ricerca su algoritmi in grado di apprendere giochi complessi senza pre-addestramento supervisionato, seguendo la linea di AlphaZero.