PR #3715 Replace deprecated API `jax.tree_map`

Replace deprecated API `jax.tree_map` #3715

ghost merged 1 commit into main from test_610040067

copybara-service force pushed 2 years ago

copybara-service force pushed 2 years ago

Replace deprecated API `jax.tree_map`

83e74665

copybara-service force pushed to 83e74665 2 years ago

copybara-service closed this 2 years ago

copybara-service deleted the test_610040067 branch 2 years ago

Reviewers

No reviews

Assignees

No one assigned

Labels

None yet

Milestone

No milestone