make per_feature_regularizer compatible