mctx 0.0.5

What's Changed

• Automatically determine num_actions and num_chance_outcomes in stochastic_muzero_policy (thanks Carlos Martin).
• Explicitly use int32 for the argmax output even when using jax_enable_x64.

Full Changelog: v0.0.3...v0.0.5

mctx 0.0.4

What's Changed

• Automatically determine num_actions and num_chance_outcomes in stochastic_muzero_policy (thanks Carlos Martin).
• Explicitly use int32 for the argmax output even when using jax_enable_x64.

Full Changelog: v0.0.3...v0.0.4

mctx 0.0.3

What's Changed

• Add experimental support for Stochastic MuZero.
• Fix pylint and pytype warnings.

Full Changelog: v0.0.2...v0.0.3

mctx 0.0.2

What's Changed

• Add a link to a0-jax on Connect Four, Gomoku, and Go.
• Improve debugging by avoiding an unused NaN.
• Use jax.tree_util to avoid deprecation warnings.

Full Changelog: https://github.com/deepmind/mctx/commits/v0.0.2