voiage.core.gpu_acceleration.gpu_jit_compile

gpu_jit_compile

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

JIT compile a function using the specified or default GPU backend.

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

Returns

JIT compiled function

Parameters:

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

Returns: Callable[..., object]