Custom Loss as nn.Module (Focal Loss) | AlgoML Prep