flax
Use `jax.tree_util.tree_map` instead of deprecated `jax.tree_map`.
#3714

Merged

Use `jax.tree_util.tree_map` instead of deprecated `jax.tree_map`. #3714

ghost merged 1 commit into main from test_609937836

copybara-service force pushed 1 year ago

Use `jax.tree_util.tree_map` instead of deprecated `jax.tree_map`.

0797d053

copybara-service force pushed to 0797d053 1 year ago

copybara-service closed this 1 year ago

copybara-service deleted the test_609937836 branch 1 year ago

Reviewers

No reviews

Assignees

No one assigned

Labels

None yet

Milestone

No milestone