rlax: Replace rlax categorical cross entropy computation with distrax components.

google-deepmind / rlax

https://rlax.readthedocs.io

Apache License 2.0

1.23k stars 85 forks source link

Closed copybara-service[bot] closed 2 years ago

copybara-service[bot] commented 2 years ago

rlax: Replace rlax categorical cross entropy computation with distrax components.