"""SimpleP2G pretrained prediction heads."""