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

Commits

Commits on Aug 21, 2026