lowrollr / turbozero

fast + parallel AlphaZero in JAX
Apache License 2.0
85 stars 7 forks source link

allow for running w/ multiple gpus and provide an example #9

Closed lowrollr closed 8 months ago

lowrollr commented 10 months ago
          And i need use mult gpus, I don't see something like pmap in code. Need a example that how to train use mult gpus.

Originally posted by @Nightbringers in https://github.com/lowrollr/turbozero/issues/8#issuecomment-1925039705

lowrollr commented 8 months ago

added with #12