Custom nn.Module | AlgoML Prep