flax
[JAX] Fix code that compares PartitionSpecs with tuples.
#5410
Merged

Commits
  • [JAX] Fix code that compares PartitionSpecs with tuples.
    a-googler committed 98 days ago
Loading