3370 and custom derivative rules.
#405

Merged

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

copybara-service merged 1 commit into master from test_326119178

google-cla added cla: yes

Improve partial_eval_by_shape compatibility with https://github.com/g…

46e9c0b2

copybara-service force pushed to 46e9c0b2 5 years ago

copybara-service merged 46e9c0b2 into master 5 years ago

copybara-service deleted the test_326119178 branch 5 years ago

Reviewers

No reviews

Assignees

No one assigned

Labels

cla: yes

Milestone

No milestone