flax
Update libraries to use JAX's limited (and ill-advised) trace-state-querying APIs rather than depending on JAX's deeper internals, which are about to change.
#4225

Merged

Update libraries to use JAX's limited (and ill-advised) trace-state-querying APIs rather than depending on JAX's deeper internals, which are about to change. #4225

copybara-service merged 1 commit into main from test_678351335

copybara-service force pushed 1 year ago

Update libraries to use JAX's limited (and ill-advised) trace-state-q…

fc19c5dc

copybara-service force pushed to fc19c5dc 1 year ago

copybara-service merged fc19c5dc into main 1 year ago

copybara-service deleted the test_678351335 branch 1 year ago

Reviewers

No reviews

Assignees

No one assigned

Labels

None yet

Milestone

No milestone