Skip to content

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

Open
copybara-service[bot] wants to merge 1 commit into
mainfrom
test_967945926
Open

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

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

3bd3584
Select commit
Loading
Failed to load commit list.
Google CLA / cla/google succeeded Aug 21, 2026 in 15s

✅ 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:

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

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