search

huangeddie / MuZeroGoJax

Mu Zero Go implemented with JAX and GoJAX
MIT License
9 stars 0 forks source link

Use Jax mixed precision library #279

Closed huangeddie closed 1 year ago

huangeddie commented 1 year ago

https://dm-haiku.readthedocs.io/en/latest/api.html#module-haiku.mixed_precision,

Split norm policy from model policy