search

huangeddie / MuZeroGoJax

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

Buffer donation on multi_train_step_fn #262

Closed huangeddie closed 1 year ago

huangeddie commented 1 year ago

https://jax.readthedocs.io/en/latest/faq.html#buffer-donation