PyTorch Wrapper: Unleashing the Power of Neural Networks