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