Sparse query masks

Rename `sparse_feature_masks` to `feature_sets`