Hi,
I am not sure to follow, ModuleList is a torch.nn construct and unrelated to this discussion about custom Functions.
Maybe open a new topic with a code sample that shows what you’re trying to achieve?