code: def model_configure(*args, **kwargs):
      kwargs["module"] = "trax.models"
      return gin.external_configurable(*args, **kwargs)
  
file_path: trax/trax/models/__init__.py
imports: import gin
  from trax.models import atari_cnn
  from trax.models import mlp
  from trax.models import neural_gpu
  from trax.models import resnet
  from trax.models import rl
  from trax.models import rnn
  from trax.models import transformer
  from trax.models.reformer import reformer
  from trax.models.research import bert
  from trax.models.research import configurable_transformer
  from trax.models.research import hourglass
  from trax.models.research import layerdrop_transformer
  from trax.models.research import rezero
  from trax.models.research import rse
  from trax.models.research import terraformer
  from trax.models.research import transformer2
label_desc: Use image node_id 1 for calling a global function with example usage: model_configure() and returns: gin