jax.extend.backend.get_backend

Contents

jax.extend.backend.get_backend#

jax.extend.backend.get_backend(platform=None)[source]#

Parameters:: platform (None | str | xla_client.Client)
Return type:: xla_client.Client