jax.lib.xla_bridge.local_device_count

jax.lib.xla_bridge.local_device_count(backend=None)[source]

Returns the number of devices addressable by this process.

Parameters

backend (Union[str, Client, None]) –

Return type

int