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

Merged

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

copybara-service merged 1 commit into main from test_898951925

copybara-service force pushed from dcc2e879 to 14424e21 97 days ago

[JAX] Fix code that compares PartitionSpecs with tuples.

dceec7d6

copybara-service force pushed from 14424e21 to dceec7d6 97 days ago

copybara-service merged dceec7d6 into main 97 days ago

copybara-service deleted the test_898951925 branch 97 days ago

Reviewers

No reviews

Assignees

No one assigned

Labels

None yet

Milestone

No milestone