Added Multi-Host TPU tutorial #9507
Open
Conversation
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
vfdev-5
force-pushed
the
docs-learn-multi-host-tpu
branch
from
4527a9f to
8c0d217
Compare
melissawm
reviewed
Jul 24, 2025
| 0 Training finished! | ||
| ``` | ||
|
|
||
| #### Profiler logs in TensorBoard |
Contributor
There was a problem hiding this comment.
I believe we want to use XProf instead of TensorBoard, but we should confirm.
Contributor
Contributor
|
Hi folks - gentle ping. If you have any feedback, we're happy to address. Thanks! |
vfdev-5
force-pushed
the
docs-learn-multi-host-tpu
branch
from
8c0d217 to
aac84ec
Compare
vfdev-5
force-pushed
the
docs-learn-multi-host-tpu
branch
from
aac84ec to
9f8c996
Compare
vfdev-5
requested review from
mikegre-google,
qihqi and
zhanyong-wan
as code owners
Contributor
|
Hi all - is this something you are still interested in? I'm happy to help bring it over the finish line if so. Thanks! |
Collaborator
|
@bhavya01 , could you take a look and recommend whether we should proceed with this? Thanks! |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
3 participants
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
This is a draft of multi-host tutorial, based on this gist: https://gist.github.com/vfdev-5/70f695e462443685a0922e79ce0ee899 and Chris Jones' mnist_xla.py code.
cc @melissawm