Login
From:
jax.readthedocs.io
(Uncensored)
subscribe
jax.vmap — JAX documentation
https://jax.readthedocs.io/en/latest/_autosummary/jax.vmap.html
links
backlinks
Roast topics
Find topics
Find it!
Vectorizing map. Creates a function which maps fun over argument axes.
