jax.lib.xla_bridge.local_device_count

Warning

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

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

previous

jax.lib.xla_bridge.get_compile_options

next

jax.lib.xla_bridge.process_index