flax
Fix CI breakages from newest jax
#4576
Merged

Fix CI breakages from newest jax #4576

copybara-service merged 1 commit into google:main from IvyZX:fix-head
IvyZX
review-notebook-app
IvyZX IvyZX force pushed from b1aab1b1 to d0ea790f 1 year ago
IvyZX IvyZX changed the title Fix doctest with correct NamedSharding arg Fix CI breakages from newest jax 1 year ago
IvyZX Fix breakages from newest jax
0f9f55d6
IvyZX IvyZX force pushed from d0ea790f to 0f9f55d6 1 year ago
levskaya
levskaya approved these changes on 2025-02-26
levskaya levskaya added pull ready
copybara-service copybara-service merged d96be6ce into main 1 year ago

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
Milestone