jax.lib.xla_bridge.get_backend#

jax.lib.xla_bridge.get_backend(platform=None)[source]#
Parameters:

platform (Union[None, str, Client]) –

Return type:

Client