Actions: google/jax
Actions
Showing runs from all workflows
35,555 workflow runs
35,555 workflow runs
ffi_call
function with a similar signature to pure_callback
.
CI
#53839:
Pull request #21743
synchronize
by
copybara-service
bot
pxla.shard_arg
batch calls to xc.copy_array_to_devices_with_sharding
CI
#53835:
Pull request #21736
synchronize
by
copybara-service
bot