->mutable_optimizer_options()
        ->set_global_jit_level(OptimizerOptions::ON_2);
  }
  if (!options.session_name.empty()) {
    session_options.config.mutable_experimental()
        ->mutable_session_metadata()
        ->set_name(options.session_name);
  }

  // Call AddDevices to register the XLA devices.
  //
  // It may be worth refactoring out XlaOpRegistry::RegisterCompilationDevice to