hr0nix/omega

Use MCTS-in-JAX

ipsec opened this issue · 3 comments

Thanks for the suggestion! The problem is though, this repo uses Stochastic MuZero, which needs a support for chance nodes in MCTS, something that mctx currently lacks. If they ever support it, I'd be happy to switch.

ipsec commented

Great

ipsec commented

Hi Boris,

Mctx now support stochastic muzero.

Commit here