search

huangeddie / MuZeroGoJax

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

Fix log_softmax to float32 #280

Closed huangeddie closed 1 year ago

huangeddie commented 1 year ago

https://github.com/google/flax/issues/677#issue-751127353