Vectorize

Returns an object that acts like pyfunc, but takes arrays as input.

Abstract Signature:

Vectorize(pyfunc: Callable)

JAX (Core)

API: jnp.vectorize
Strategy: Direct Mapping
Official Docs ↗

NumPy

API: numpy.vectorize
Strategy: Direct Mapping
Official Docs ↗

Keras

API: keras.ops.vectorize
Strategy: Direct Mapping
Official Docs ↗

Flax NNX

API: jax.numpy.vectorize
Strategy: Direct Mapping
Official Docs ↗