` `init_method`
#5438

Merged

[PJRT] Support `torchrun` with `pjrt://` `init_method` #5438

will-cromar merged 6 commits into master from wcromar/tpu-torchrun

Support torchrun with `pjrt://` `init_method`

dfc83e71

move import

feb13596

fix error

144b6afd

Fix NameError

f6341e5a

Fix path

fb458be4

Remove from TPU CI

7679ccdd

will-cromar marked this pull request as ready for review 2 years ago

will-cromar requested a review from

JackCaoG 2 years ago

JackCaoG approved these changes on 2023-08-14

will-cromar merged 9db8fac4 into master 2 years ago

Reviewers

JackCaoG

Assignees

No one assigned

Labels

None yet

Milestone

No milestone