Skip to content

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

Open
jakevdp wants to merge 1 commit intojax-ml:mainfrom jakevdp:vmap-convert-args
+19-2

Commits

Commits on Jan 10, 2025