PR #1175 Flax tests: don't hardcode number of devices

Flax tests: don't hardcode number of devices #1175

patrickvonplaten merged 1 commit into main from flax-tests-devices

Flax tests: don't hardcode number of devices.

fad01440

pcuenca requested a review from

patrickvonplaten 3 years ago

pcuenca requested a review from

anton-l 3 years ago

patrickvonplaten approved these changes on 2022-11-07

patrickvonplaten merged af279434 into main 3 years ago

kashif deleted the flax-tests-devices branch 2 years ago

Reviewers

patrickvonplaten

anton-l

Assignees

No one assigned

Labels

None yet

Milestone

No milestone