gpu_engine = pl.GPUEngine(
    device=0,
    raise_on_fail=True,
)