Skip to content
CI - Cloud TPU (presubmit)

Make make_mesh take visible_axes, hidden_axes and collective_axes as parameters instead of axis_types to make it a more cleaner API. #1781

Sign in to view logs

Make make_mesh take visible_axes, hidden_axes and collective_axes as parameters instead of axis_types to make it a more cleaner API.

Make make_mesh take visible_axes, hidden_axes and collective_axes as parameters instead of axis_types to make it a more cleaner API. #1781

Run time

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