saiunit.set_default_backend

saiunit.set_default_backend#

saiunit.set_default_backend(name)[source]#

Set the default backend used when input backend is ambiguous.

Parameters:

name (Optional[Literal['numpy', 'jax', 'cupy', 'torch', 'dask', 'ndonnx']]) – Pass None to clear the default. With no default, the tie-breaker prefers JAX when installed and falls back to NumPy otherwise.

Return type:

None