Composite(
  GlobalTypeRule(  # on all layers
    Conv => ZPlusRule(),
    Dense => EpsilonRule{Float32}(1.0f-6),
    Union{typeof(Flux.dropout), AlphaDropout, Dropout} => PassRule(),
    Union{typeof(flatten), typeof(MLUtils.flatten)} => PassRule(),
  ),
)
