jax.process_count

jax.process_count(backend=None)

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

参数:

backend (str | xla_client.Client | None)

返回类型:

int