code: def make_builder(config: RainbowConfig):
      """Returns a DQNBuilder with a pre-built loss function."""
      loss_fn = losses.PrioritizedCategoricalDoubleQLearning(
          discount=config.discount,
          importance_sampling_exponent=config.importance_sampling_exponent,
          max_abs_reward=config.max_abs_reward,
      )
      return builder.DQNBuilder(config, loss_fn=loss_fn)
  
file_path: acme/acme/agents/jax/dqn/rainbow.py
imports: import dataclasses
  from typing import Callable
  from acme import specs
  from acme.agents.jax.dqn import actor
  from acme.agents.jax.dqn import builder
  from acme.agents.jax.dqn import config
  from acme.agents.jax.dqn import losses
  from acme.jax import networks
  from acme.jax import utils
  import rlax
label_desc: Use image node_id 3 for calling a global function with example usage: make_builder(config) and returns: builder