The type should be "callable, optional". I'd prefer `"tanh"` (as a string) to `None` for the default value.