flax
make flax work with upcoming JAX change to tree_map (being more careful about
#3976
Merged

make flax work with upcoming JAX change to tree_map (being more careful about #3976

ghost merged 1 commit into main from test_641067882
copybara-service
copybara-service copybara-service force pushed 1 year ago
mattjj make flax work with upcoming JAX change to tree_map (being more caref…
e9bde53c
copybara-service copybara-service force pushed to e9bde53c 1 year ago
copybara-service copybara-service closed this 1 year ago
copybara-service copybara-service deleted the test_641067882 branch 1 year ago

Login to write a write a comment.

Login via GitHub

Reviewers
No reviews
Assignees
No one assigned
Labels
Milestone