code: def main(_):
      experiment_config = build_experiment_config()
  
      if not RUN_DISTRIBUTED.value:
          raise NotImplementedError(
              "Single threaded experiment not supported."
          )
  
      inference_server_config = (
          inference_server_lib.InferenceServerConfig(
              batch_size=64,
              update_period=400,
              timeout=datetime.timedelta(
                  seconds=1,
              ),
          )
      )
      num_inference_servers = math.ceil(
          NUM_ACTORS.value / (128 * NUM_ACTORS_PER_NODE.value),
      )
  
      program = experiments.make_distributed_experiment(
          experiment=experiment_config,
          num_actors=NUM_ACTORS.value,
          num_learner_nodes=NUM_LEARNERS.value,
          num_actors_per_node=NUM_ACTORS_PER_NODE.value,
          num_inference_servers=num_inference_servers,
          inference_server_config=inference_server_config,
      )
      lp.launch(
          program,
          xm_resources=lp_utils.make_xm_docker_resources(
              program,
          ),
      )
  
file_path: acme/examples/baselines/rl_discrete/run_muzero.py
imports: import datetime
  import math
  from absl import flags
  from acme import specs
  from acme.agents.jax import muzero
  import helpers
  from absl import app
  from acme.jax import experiments
  from acme.jax import inference_server
  from acme.utils import lp_utils
  import dm_env
  import launchpad
label_desc: Use image node_id 2 for calling a global function with example usage: main(_) without return types