Thanks to all, so basically this is still impossible. The sequential code needs a slight modification in it’s forward to consider mutli-output modules:
def forward(self, input): 
    for module in self._modules.values():
        input = module(input)
        if isinstance(input, tuple):
            input = input[0] 
    return input

I filed this issue 17 as well.