fix floatX in whiten_transform