I can’t think of a breaking use case at the moment, as hooks, nn.DataParallel, etc. should all work in your approach. This is of course not a valid verification, so let me think about some more use cases.