jax.extend.backend.register_backend_factory#

jax.extend.backend.register_backend_factory(name, factory, *, priority=0, fail_quietly=True, experimental=False, make_topology=None, c_api=None)[source]#
Parameters:

  • name (str)

  • factory (BackendFactory)

  • priority (int)

  • fail_quietly (bool)

  • experimental (bool)

  • make_topology (TopologyFactory | None)

  • c_api (Any | None)

Return type:

None