jax.lib.xla_bridge.get_backend

Contents

jax.lib.xla_bridge.get_backend#

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

Parameters: platform (Union[str, Client, None]) –
Return type: Client