[Chess] Accelerate chess_utils.py import (jnp => np) (#1170)

[Chess] Accelerate chess_utils.py import (jnp => np) (#1170) #319