voiage.core.gpu_acceleration.gpu_parallelize

gpu_parallelize

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

Parallelize a function across multiple GPUs using the specified or default backend.

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

Returns

Parallelized function

Parameters:

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

Returns: Callable[..., object]