jax.process_count

jax.process_count(backend=None)

返回与后端关联的 JAX 进程数。

参数：

backend (str | xla_client.Client | None)

返回类型：

int