huangeddie/GoJAX
The board game Go implemented in JAX for fast game processing and machine learning training.
PythonMIT
Issues
- 0
optimize next_states
#38 opened by huangeddie - 0
print_state
#43 opened by huangeddie - 0
- 0
- 0
action1d_to_indicator
#41 opened by huangeddie - 0
Add get_children function
#40 opened by huangeddie - 0
Benchmark next_states between indicator action implementation and action 1D implementation
#37 opened by huangeddie - 0
Benchmark changing turns
#36 opened by huangeddie - 0
Make horizontal index in get_pretty_string alphabetical letters instead of numbers
#35 opened by huangeddie - 0
Benchmark different solutions to _paint_fill
#34 opened by huangeddie - 0
Benchmark smaller functions
#26 opened by huangeddie - 0
Benchmark on CPU / GPU / TPU
#20 opened by huangeddie - 0
- 0
Refactor tests
#31 opened by huangeddie - 0
- 0
Remove all array initialization in functions
#28 opened by huangeddie - 1
- 0
- 0
Improve unit tests with Chex
#5 opened by huangeddie - 0
Parameterize the test cases
#16 opened by huangeddie - 0
Use two space indent
#27 opened by huangeddie - 0
Add random categorical sampling of actions
#22 opened by huangeddie - 0
Add benchmark code
#24 opened by huangeddie - 0
Experiment with smaller dtypes for lax.conv
#23 opened by huangeddie - 1
Profile GoJAX benchmark
#25 opened by huangeddie - 0
Ask for credit in README if used
#18 opened by huangeddie - 0
Cast inputs to lax.conv as floats
#21 opened by huangeddie - 0
Replace jsp.signal.convolution with lax.conv
#19 opened by huangeddie - 0
Link old GymGo to new GoJAX
#17 opened by huangeddie - 0
Make gojax the source directory
#15 opened by huangeddie - 0
Resolve pylints
#14 opened by huangeddie - 0
Add GutHub CI
#13 opened by huangeddie - 0
Add areas to get_pretty_string
#12 opened by huangeddie - 0
Add type hints
#11 opened by huangeddie - 0
Remove the inner go module
#9 opened by huangeddie - 0
- 0
Autogenerated docs
#7 opened by huangeddie - 0
New `swap_perspective` function
#1 opened by huangeddie - 0
get_winner function
#3 opened by huangeddie - 0
get_area_size function
#6 opened by huangeddie - 0
New get_areas function
#2 opened by huangeddie - 0
test remove multiple groups
#4 opened by huangeddie