code: def afx_small_bfloat16():
      """Small transformer model with small batch size for fast step times."""
      hparams = afx_small()
      hparams.weight_dtype = "bfloat16"
      hparams.activation_dtype = "bfloat16"
      return hparams
  
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  from tensor2tensor.models import transformer
  from tensor2tensor.utils import registry
label_desc: Use image node_id 24 for calling a global function with example usage: afx_small_bfloat16() and returns: hparams