def METHOD_NAME(self, in_features: int) -> nn.Module:
    r"""
    Return the parameter projection layer that maps the input to the appropriate parameters of the distribution.
    """
    return ParameterProjection(
        in_features=in_features,
        args_dim=self.args_dim,
        domain_map=LambdaLayer(self.domain_map),
    )