jax.process_count

jax.process_count(backend=None)

傳回與後端相關聯的 JAX 處理程序數量。

參數:

backend (str | xla_client.Client | None)

回傳類型:

int