Actions: jax-ml/jax
Actions
1,866 workflow runs
1,866 workflow runs
reshard
API in experimental. Currently for sharding_in_types we have 2 APIs: mesh_cast
and reshard
. Both work in sharding_in_types mode and affect the sharding of the aval. Following are the semantics of both:
CI - Cloud TPU (presubmit)
#1754:
Pull request #25978
synchronize
by
copybara-service
bot