jax.local_device_count

Warning

This page was created from a pull request (#9655).

jax.local_device_count¶

jax.local_device_count(backend=None)[source]¶

Returns the number of devices addressable by this process.

Parameters

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

Return type

int

previous

jax.device_count

next

jax.process_count