Skip to content
CI - Cloud TPU (presubmit)

Put abstract_mesh on every eqn so that we can preserve it during eval_jaxpr and check_jaxpr roundtrip. #133

Sign in to view logs

Put abstract_mesh on every eqn so that we can preserve it during eval_jaxpr and check_jaxpr roundtrip.

Put abstract_mesh on every eqn so that we can preserve it during eval_jaxpr and check_jaxpr roundtrip. #133

Run time

Learn about OS pricing on GitHub Actions
Job Run time
TPU test (jaxlib=head, v5e-8)
10m 52s
10m 52s