Support device_put of an uncommitted array to a single global device in McJAX.
#12383
bazel_cpu_presubmit.yml
on: pull_request
Build jax artifact
/
jax, linux x86, py 3.12, clone main XLA=0
2m 46s
Matrix: build-jaxlib-artifact
Matrix: run_tests