search

google-research / t5x

Apache License 2.0
2.58k stars 296 forks source link

jax.tree.map->jax.tree_map #1535

Open jntdst opened 3 months ago

jntdst commented 3 months ago

Jax does not have tree object, it is changed to tree_map.

google-cla[bot] commented 3 months ago

Thanks for your pull request! It looks like this may be your first contribution to a Google open source project. Before we can look at your pull request, you'll need to sign a Contributor License Agreement (CLA).

View this failed invocation of the CLA check for more information.

For the most up to date status, view the checks section at the bottom of the pull request.