Merge "Add flag for Normalized PWLE feature" into main