google-deepmind / mctx

Monte Carlo tree search in JAX
Apache License 2.0
2.31k stars 188 forks source link

Questions on library features #56

Closed AI3EL closed 1 year ago

AI3EL commented 1 year ago

Hi, thank you for your work.

I am considering using it but I had a few questions before diving deeper:

Many Thanks

fidlej commented 1 year ago

Quick answers:

  1. No support for an array of discounts.
  2. The MCTS is synchronous. Parallel computation is done by using a batch size.
  3. CPU works but would be probably slower.
  4. I do not see a blocker. You would have to benchmark it to check that the speed is OK.
  5. You would not benefit from the parallel computation with batch_size=1. On CPU, you can instead write an optimized MCTS in C++ or Rust.