jax.local_device_count

jax.local_device_count(backend=None)

Returns the number of devices addressable by this process.

Parameters

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

Return type

int