Skip to content
Back to pull request #2173

Fix xla_pmap_p import for JAX versions that removed pmap #4469

Sign in to view logs

Fix xla_pmap_p import for JAX versions that removed pmap

Fix xla_pmap_p import for JAX versions that removed pmap #4469

test-inference (3.11)

test-inference (3.11)
succeeded May 2, 2026 in 46m 4s