code: def autoregressive_model(c: Configs):
      """
      ### Initialize the auto-regressive model
      """
      m = AutoregressiveModel(c.n_tokens, c.d_model, c.transformer)
      return m.to(c.device)
  
file_path: annotated_deep_learning_paper_implementations/labml_nn/transformers/switch/experiment.py
imports: import torch
  import torch.nn
  from labml import experiment, tracker
  from labml.configs import option
  from labml_helpers.module import Module
  from labml_helpers.train_valid import BatchIndex
  from labml_nn.experiments.nlp_autoregression import NLPAutoRegressionConfigs
label_desc: Use image node_id 1 for calling a global function with example usage: autoregressive_model(c) and returns: m