trainer: add Reusable ClusterTrainingRuntime for jax on tpu guide

[siyuanfoundation]

Description of Changes

Followup for #4343 to add example of TPU ClusterTrainingRuntime

Related Issues

Related: kubeflow/trainer#3183

Checklist

• You have signed off your commits
• Ensure you follow best practices from our contributing guide.
• (for big changes) I will post screenshots of the changes in a PR comment

[google-oss-prow]

Hi @siyuanfoundation. Thanks for your PR.

I'm waiting for a kubeflow member to verify that this patch is reasonable to test. If it is, they should reply with /ok-to-test on its own line. Until that is done, I will not automatically test new commits in this PR, but the usual testing commands by org members will still work. Regular contributors should join the org to skip this step.

Once the patch is verified, the new status will be reflected by the ok-to-test label.

I understand the commands that are listed here.

Details

Instructions for interacting with me using PR comments are available here. If you have questions or suggestions related to my behavior, please file an issue against the kubernetes/test-infra repository.

[github-actions]

🚫 This command cannot be processed. Only organization members or owners can use the commands.

[siyuanfoundation]

/cc @andreyvelich
Can you take a look at the PR? Thanks!

[Arhell]

/ok-to-test

[andreyvelich]

Thank you @siyuanfoundation!
/lgtm
/approve

[google-oss-prow]

[APPROVALNOTIFIER] This PR is APPROVED

This pull-request has been approved by: andreyvelich

The full list of commands accepted by this bot can be found here.

The pull request process is described here

Details

Needs approval from an approver in each of these files:

• content/en/docs/components/trainer/OWNERS [andreyvelich]

Approvers can indicate their approval by writing /approve in a comment
Approvers can cancel approval by writing /approve cancel in a comment