https://hgpu.org/?p=28102
Pgx: Hardware-accelerated parallel game simulation for reinforcement learning