google-deepmind/mctx v0.0.4

mctx 0.0.4

Repository: google-deepmind/mctx

Tag: v0.0.4

Published: 2023-11-24T10:49:59Z

Prerelease: no

Release notes:

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: https://github.com/google-deepmind/mctx/compare/v0.0.3...v0.0.4