GitHub repository: lockwo/pgx
Description: ♟️ Vectorized RL game environments in JAX