Actions: jax-ml/jax
Actions
1,788 workflow runs
1,788 workflow runs
mgpu
in wgmma.py
CI - Cloud TPU (presubmit)
#1787:
Pull request #25999
synchronize
by
copybara-service
bot
mgpu
in wgmma.py
CI - Cloud TPU (presubmit)
#1786:
Pull request #25999
opened
by
copybara-service
bot
make_mesh
take visible_axes
, hidden_axes
and collective_axes
as parameters instead of axis_types
to make it a more cleaner API.
CI - Cloud TPU (presubmit)
#1782:
Pull request #25997
synchronize
by
copybara-service
bot
make_mesh
take visible_axes
, hidden_axes
and collective_axes
as parameters instead of axis_types
to make it a more cleaner API.
CI - Cloud TPU (presubmit)
#1781:
Pull request #25997
opened
by
copybara-service
bot