flax
Improve partial_eval_by_shape compatibility with https://github.com/google/jax/pull/3370 and custom derivative rules.
#403
Merged

Improve partial_eval_by_shape compatibility with https://github.com/google/jax/pull/3370 and custom derivative rules. #403

copybara-service merged 1 commit into master from test_326015295
copybara-service
google-cla google-cla added cla: yes
codecov-commenter
copybara-service copybara-service force pushed 6 years ago
mattjj Improve partial_eval_by_shape compatibility with https://github.com/g…
472b82a2
copybara-service copybara-service force pushed to 472b82a2 6 years ago
copybara-service copybara-service merged 472b82a2 into master 6 years ago
copybara-service copybara-service deleted the test_326015295 branch 6 years ago

Login to write a write a comment.

Login via GitHub

Reviewers
No reviews
Assignees
No one assigned
Labels
Milestone