ENH: enable low rank whitening