GPU 向量化

深度分析

Mahjax：以 JAX 向量化與多 GPU 加速 Riichi 麻將模擬

Mahjax 是以 JAX 實作、完全向量化的 Riichi 麻將模擬環境，目標是在 GPU 上實現大規模並行 rollout，降低傳統 CPU 模擬器的瓶頸。設計上採用不可變狀態資料結構、把控制流程改寫為矩陣運算、並對計算密集的役（Yaku）判定做快取化處理。