Skip to content

Update run_connect_example.py to demonstrate a TPU workload using JAX sharding - #313

Merged
copybara-service[bot] merged 1 commit into
mainfrom
test_967945926
Aug 24, 2026
Merged

Update run_connect_example.py to demonstrate a TPU workload using JAX sharding#313
copybara-service[bot] merged 1 commit into
mainfrom
test_967945926

Update run_connect_example.py to demonstrate a TPU workload using JAX…

9c90fd6
Select commit
Loading
Failed to load commit list.
Google CLA / cla/google succeeded Aug 24, 2026 in 14s

✅ All contributors are covered under a CLA with Google

See https://cla.developers.google.com/ for more info about Google's Contributor License Agreement (CLA).

ℹ️ Googlers: Go here to view more details and manage scans for this pull request.

Details

The following contributors were found for this pull request:

9c90fd6 PR Opener: @copybara-service[bot]
9c90fd6 Author: @guptaaka <a***u​@google.com>

(Only the first commit for a unique contributor is listed.)