Skip to content
CI - Cloud TPU (presubmit)

Allow P.UNCONSTRAINED in out_shardings at top level jit. This is required for sharding in types to work properly when out_avals contain UNCONSTRAINED specs. #109

Sign in to view logs

Allow P.UNCONSTRAINED in out_shardings at top level jit. This is required for sharding in types to work properly when out_avals contain UNCONSTRAINED specs.

Allow P.UNCONSTRAINED in out_shardings at top level jit. This is required for sharding in types to work properly when out_avals contain UNCONSTRAINED specs. #109

TPU test (jaxlib=head, v5e-8)

TPU test (jaxlib=head, v5e-8)
succeeded Dec 17, 2024 in 5m 55s