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