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

lint (3.11)

lint (3.11)
succeeded May 2, 2026 in 3m 58s