aglib/meliad/transformer/configs/recurrent/states_256.gin

# Apply after the gate config.

transformer_layer.TransformerLayer:
  recurrent_num_states = 256