code: def _alibi_mha(c: TransformerConfigs):
      """
      Create an ALiBi attention module
      """
      return AlibiMultiHeadAttention(
          c.n_heads, c.d_model, dropout_prob=c.dropout
      )
  
file_path: annotated_deep_learning_paper_implementations/labml_nn/transformers/alibi/experiment.py
imports: import torch
  from torch.utils.data import DataLoader
  from labml import experiment, tracker
  from labml.configs import option, calculate
  from labml_helpers.datasets.text import SequentialUnBatchedDataset
  from labml_nn.transformers.alibi import AlibiMultiHeadAttention
  from labml_nn.experiments.nlp_autoregression import transpose_batch
  from labml_nn.transformers import TransformerConfigs
  from labml_nn.transformers.gpt import Configs
label_desc: Use image node_id 1 for calling a global function with example usage: _alibi_mha(c) and returns: AlibiMultiHeadAttention