def METHOD_NAME(self) -> Dict[str, Dict[str, Any]]:
    """Represent ModelConfig as a dict.
    This builds on `BaseModel.dict()` but wraps the output in a single-key
    dictionary to make it unambiguous to identify model arguments that are
    themselves models.
    """
    config_dict = self.dict()
    config_dict = traverse_and_apply(
        obj=dict(**config_dict), fn=self._parse_torch
    )
    return {self.__class__.__name__: config_dict}