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