jax.process_count# jax.process_count(backend=None)[源代码][源代码]# 返回与后端关联的JAX进程数。 参数: backend (str | xla_client.Client | None) 返回类型: int