ray.util.accelerators.tpu.get_current_pod_worker_count#

ray.util.accelerators.tpu.get_current_pod_worker_count() → int | None[source]#

Count the number of workers associated with the TPU pod that the worker belongs to.

Returns:: The total number of workers in the TPU pod. Returns None if the worker is not part of a TPU pod.

PublicAPI (alpha): This API is in alpha and may change before becoming stable.