Vectorizing map. Creates a function which maps fun over argument axes.| jax.readthedocs.io
🔪 Pure functions#| jax.readthedocs.io
