rlax: Upstream pytree transpose helper functions to rlax. - Githubissues

google-deepmind / rlax

https://rlax.readthedocs.io

Apache License 2.0

1.24k stars 85 forks source link

rlax: Upstream pytree transpose helper functions to rlax. #104

Closed copybara-service[bot] closed 1 year ago

copybara-service[bot] commented 1 year ago

rlax: Upstream pytree transpose helper functions to rlax.