Solved by justusschock in post #5 
              
              
                From my understandingof the code, this is needed for torch.jit only to create a static computation graph, meaning that the function calls to/from this class will be lazily compiled and inlined