jax_muzero icon indicating copy to clipboard operation
jax_muzero copied to clipboard

An implementation of MuZero in JAX.

Results 1 jax_muzero issues
Sort by recently updated
recently updated
newest added

Hello! I've noticed that the gradient is being scaled by 0.5 after every model step: https://github.com/Hwhitetooth/jax_muzero/blob/b8ab36251d22e0246d61514841ba17a22f4b2a36/algorithms/agents.py#L116-L120 Can you clarify the motivation for that? Are you aware of any experimental results...