Skip to content
CI - Cloud TPU (presubmit)

[sharding_in_types] Add jax_varying_axes_in_types config so that we can start adding varying_manual_axes to ShapedArray. #5788

Sign in to view logs

[sharding_in_types] Add jax_varying_axes_in_types config so that we can start adding varying_manual_axes to ShapedArray.

[sharding_in_types] Add jax_varying_axes_in_types config so that we can start adding varying_manual_axes to ShapedArray. #5788

Triggered via pull request March 12, 2025 00:55
@copybara-servicecopybara-service[bot]
opened #27080
test_735935837
Status Success
Total duration 10m 33s
Artifacts –

cloud-tpu-ci-presubmit.yml

on: pull_request
Matrix: build-jax-artifacts
TPU test (jaxlib=head, v5e-8)  /  Pytest TPU (v5e-8, Python 3.10, libtpu=nightly)
7m 2s
TPU test (jaxlib=head, v5e-8) / Pytest TPU (v5e-8, Python 3.10, libtpu=nightly)
Fit to window
Zoom out
Zoom in