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 342 days ago
IvyZX IvyZX changed the title Fix doctest with correct NamedSharding arg Fix CI breakages from newest jax 342 days ago
IvyZX Fix breakages from newest jax
0f9f55d6
IvyZX IvyZX force pushed from d0ea790f to 0f9f55d6 342 days ago
levskaya
levskaya approved these changes on 2025-02-26
levskaya levskaya added pull ready
copybara-service copybara-service merged d96be6ce into main 342 days ago

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
Milestone