def _extract_parameters_from_exir_program(
    exported_program,  # torch.export.ExportedProgram
) -> Dict[str, torch.Tensor]:
    """
    Given:
        exported_program: torch.export.ExportedProgram
    Return:
        parameters: dictionary mapping names to torch parameter tensors
    """
    parameters = {}
    for name, parameter in zip(
        exported_program.graph_signature.parameters, exported_program.parameters()
    ):
        if not isinstance(parameter, torch.Tensor):
            raise NotImplementedError(
                f"Only torch.Tensor parameter handled yet, but got {type(parameter)}"
            )
        parameters[name] = parameter
    return parameters

    Given:
        exported_program: torch.export.ExportedProgram
    Return:
        parameters: dictionary mapping names to torch parameter tensors
    
coremltools/converters/mil/frontend/torch/exir_utils.py