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. #1815

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. #1815

TPU test (jaxlib=head, v5e-8)

TPU test (jaxlib=head, v5e-8)
succeeded Jan 21, 2025 in 13m 36s