Skip to content

voiage.core.gpu_acceleration.gpu_vectorize

gpu_vectorize([positional or keyword] func: Callable[..., object] = None, [positional or keyword] backend: str | None = None) -> Callable[..., object]

Vectorize a function using the specified or default GPU backend.

Args: func: Function to vectorize backend: GPU backend to use (‘jax’, ‘cupy’, ‘torch’, or None for auto-detect)

Vectorized function

Parameters:

  • func Callable[..., object]
  • backend str | None (default: None)

Returns: Callable[..., object]