Fix CI breakages from newest jax #4576
IvyZX
force pushed
from
b1aab1b1
to
d0ea790f
342 days ago
IvyZX
changed the title Fix doctest with correct NamedSharding arg Fix CI breakages from newest jax 342 days ago
Fix breakages from newest jax
0f9f55d6
IvyZX
force pushed
from
d0ea790f
to
0f9f55d6
342 days ago
levskaya
approved these changes
on 2025-02-26
Assignees
No one assigned
Login to write a write a comment.
Login via GitHub