jax.local_device_count

jax.local_device_count(backend=None)

返回此进程可访问的设备数量。

参数：

backend (str | xla_client.Client | None | None)

返回类型：

int