PyTorch + Follow Up (Custom for Qualcomm)