Skip to content

[pmap] In-line definitions of jax.device_put_sharded and jax.device_put_replicated. #3056

[pmap] In-line definitions of jax.device_put_sharded and jax.device_put_replicated.

[pmap] In-line definitions of jax.device_put_sharded and jax.device_put_replicated. #3056

Annotations

3 errors and 1 warning

Pytest 3.11 on ubuntu-latest jax=newest

cancelled Apr 10, 2026 in 1m 10s