jax.process_count

Warning

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

jax.process_count¶

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

Returns the number of JAX processes associated with the backend.

Parameters

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

Return type

int

previous

jax.local_device_count