Actions: jax-ml/jax
Actions
5,720 workflow runs
5,720 workflow runs
jax_varying_axes_in_types
config so that we can start adding varying_manual_axes
to ShapedArray.
CI - Cloud TPU (presubmit)
#5788:
Pull request #27080
opened
by
copybara-service
bot
get_ty
to typeof
which is an alias of get_aval
CI - Cloud TPU (presubmit)
#5785:
Pull request #27078
synchronize
by
copybara-service
bot
get_ty
to typeof
which is an alias of get_aval
CI - Cloud TPU (presubmit)
#5784:
Commit 3a26804
pushed
by
copybara-service
bot
get_ty
to typeof
which is an alias of get_aval
CI - Cloud TPU (presubmit)
#5781:
Pull request #27078
synchronize
by
copybara-service
bot
get_ty
to typeof
which is an alias of get_aval
CI - Cloud TPU (presubmit)
#5769:
Pull request #27078
opened
by
copybara-service
bot
ParsedPartitionSpec
and preprocess
function and do a couple more cleanups
CI - Cloud TPU (presubmit)
#5768:
Pull request #26861
synchronize
by
copybara-service
bot
full
and use_mesh
outside jit did not work because the shard
passed to make_array_from_callback
was sharded on all devices instead of just 1 device.
CI - Cloud TPU (presubmit)
#5767:
Pull request #27076
synchronize
by
copybara-service
bot
full
and use_mesh
outside jit did not work becaus…
CI - Cloud TPU (presubmit)
#5766:
Commit f45cbf3
pushed
by
copybara-service
bot