Naukowcy opracowali Mahjax, specjalizowany symulator gry Mahjong zoptymalizowany dla szybkich obliczeń na GPU przy użyciu frameworku JAX. Narzędzie umożliwia efektywne szkolenie agentów reinforcement learning dzięki możliwości równoległego przetwarzania wielu gier jednocześnie. Jest to istotne dla badaczy pracujących nad algorytmami RL, gdyż Mahjong to złożona gra wymagająca strategicznego myślenia, a GPU-akceleracja drastycznie skraca czas treningów.