huangeddie / gojax Goto Github PK

Causes device writing / transfer

Would prefer action 1D

in addition to returning whether the move is invalid, it can also return the potential partial next state that sets the piece and removes any opponent pieces, and the killed pieces.

Following the JAX example

Use Chex to assert shapes

Use np.testing.assert_array_equal to assert array equality

https://jax.readthedocs.io/en/latest/profiling.html

ints and bools are unsupported

• Current
• Paint fill should use integer counter to alternate between the two stages instead of swapping them
• For loop with generous fixed iterations

Potentially expensive to update 361 elements

Should just be gojax.functions

takes in N x H x C x B x B history array and returns an integer array of shape (N, 1) with values in {-1, 0, 1}

• 1 = black wins
• 0 = tie
• -1 = white wins

Redundantly changing turns, passing setting killed pieces and ending the game.

Swaps the black and white pieces and changes the turn

Make spreadsheet

Remove the src directory prefix

Making an invalid move will simply do nothing and change the turn

Takes in action1ds and action size.

Should be JIT compatible.

Replace old function.