Solved by ptrblck in post #3 
              
              
                You can reuse parameter-less modules, as only the computation will be tracked. 
A theoretical example would be reusing nn.Add(a, b) instead of a + b.