PR #5420 Avoid usage of deprecated `jax.core` APIs.

Avoid usage of deprecated `jax.core` APIs. #5420

copybara-service wants to merge 1 commit into main from test_900270194

copybara-service force pushed from 16138856 to c4d07728 26 days ago

copybara-service force pushed from c4d07728 to 20446713 26 days ago

copybara-service force pushed from 20446713 to 11a6fe6a 26 days ago

copybara-service force pushed from 11a6fe6a to 5968b506 26 days ago

copybara-service force pushed from 5968b506 to 6a8d69a8 26 days ago

copybara-service force pushed from 6a8d69a8 to 52f95c8a 26 days ago

copybara-service force pushed from 52f95c8a to 117c052c 25 days ago

copybara-service force pushed from 117c052c to 0deab51e 25 days ago

Avoid usage of deprecated `jax.core` APIs.

c8e3120f

copybara-service force pushed from 0deab51e to c8e3120f 24 days ago

Reviewers

No reviews

Assignees

No one assigned

Labels

None yet

Milestone

No milestone