Skip to content

Tweak code example in TPU Pipelining tutorial #31482

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom
Merged
merged 1 commit into from
Sep 4, 2025
Merged

Tweak code example in TPU Pipelining tutorial #31482

merged 1 commit into from
Sep 4, 2025

Conversation

mtsokol
Copy link
Contributor

@mtsokol mtsokol commented Sep 2, 2025

For the sake of cohesiveness, out_shape should be a ShapeDtypeStruct.

@superbobry
Copy link
Collaborator

Thanks!

@google-ml-butler google-ml-butler bot added kokoro:force-run pull ready Ready for copybara import and testing labels Sep 4, 2025
@copybara-service copybara-service bot merged commit 147834d into jax-ml:main Sep 4, 2025
23 of 25 checks passed
@mtsokol mtsokol deleted the tweak-pl-tpu-pipelining-tutorial branch September 7, 2025 18:41
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Reviewers

@superbobry superbobry superbobry approved these changes

Assignees

No one assigned

Labels

kokoro:force-run pull ready Ready for copybara import and testing

Projects

None yet

Milestone

No milestone

Development

Successfully merging this pull request may close these issues.

2 participants

@mtsokol @superbobry