def set_backend(name):
  """Sets the default backend to use in Trax."""
  if name:
    _assert_valid_backend_name(name)
  global default_backend
  default_backend = name
Sets the default backend to use in Trax.
trax/fastmath/ops.py