jax.process_count

jax.process_count = <jaxlib.weakref_lru_cache.StrongLRUCache object>

Parameters:

backend (str | xla_client.Client | None)

Return type:

int