jax.process_count

jax.process_count#

jax.process_count(backend=None)[源代码][源代码]#

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

参数:

backend (str | xla_client.Client | None)

返回类型:

int