Skip to content
Back to pull request #25835

jax.vmap: convert mapped input arguments to array #1220

Sign in to view logs

jax.vmap: convert mapped input arguments to array

jax.vmap: convert mapped input arguments to array #1220

TPU test (jaxlib=head, v5e-8)

TPU test (jaxlib=head, v5e-8)
succeeded Jan 10, 2025 in 6m 24s