Review and clean up changes to nuts-rs #153
Description
The normalizing flow feature required some refactoring in nuts-rs.
We need to test that that this didn't accidentally change anything about the mass matrix sampler before we can make a release with this new feature.