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 | Client)

Return type:

Client