jax.extend.backend.get_backend# jax.extend.backend.get_backend = <jaxlib.weakref_lru_cache.StrongLRUCache object>[source]# Parameters: platform (None | str | xla_client.Client) Return type: xla_client.Client
