jax.local_devices# jax.local_devices = <jaxlib.weakref_lru_cache.StrongLRUCache object>[source]# Parameters: process_index (int | None) backend (str | xla_client.Client | None) host_id (int | None) Return type: list[xla_client.Device]
