get_backend#

t3toolbox.backend.common.get_backend(is_uniform, use_jax)#
def get_backend(
        is_uniform: bool,
        use_jax: bool,
):
Parameters:
  • is_uniform (bool)

  • use_jax (bool)