PGX provides GPU-accelerated parallel game environments (Go, Chess, Shogi, Backgammon, Poker, etc.) implemented in JAX. It vectorizes multiple game instances to enable efficient batched simulation, a key requirement for training RL agents. The library is designed to support research workflows like AlphaZero, where fast environment rollout and batched state processing are essential.