lowrollr / turbozero

fast + parallel AlphaZero in JAX
Apache License 2.0
83 stars 6 forks source link

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

Closed lowrollr closed 7 months ago

lowrollr commented 8 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 7 months ago

added with #12