jax.lib.xla_bridge.get_backend

Contents

jax.lib.xla_bridge.get_backend#

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

platform (None | str | xla_client.Client) –

Return type:

xla_client.Client