Add typing stub for Module __call__
Typecheckers like pytype currently don't recognize Module as a Callable
type when "nn.Module" is used as a dataclass attribute type, leading to
spurious typing errors, this adds a type-checking-only __call__ stub,
and moves the previous __init__ stub under the type-checking guard as
well.
