Returns the list of all JAX process indices associated with the backend.
Parameters:
backend (str | xla_client.Client | None | None) – This is an experimental feature and the API is likely to change.
Optional, a string representing the xla backend: 'cpu', 'gpu', or
'tpu'.
