Composite(
  LastLayerTypeMap(  # last layer
    Dense => EpsilonRule{Float32}(2.0f-5),
    Conv  => EpsilonRule{Float32}(0.0002f0),
 ),
  FirstLayerTypeMap(  # first layer
    Dense => AlphaBetaRule{Float32}(1.0f0, 0.0f0),
    Conv  => AlphaBetaRule{Float32}(2.0f0, 1.0f0),
 ),
)