[pmap] Avoid degraded performance under the new jax.pmap.
#1840
copybara-service[bot]Annotations
3 errors
|
build (3.11)
Process completed with exit code 1.
|
|
build (3.10)
The strategy configuration was canceled because "build._3_11" failed
|
|
build (3.10)
The operation was canceled.
|