I do have a rough notebook testing this. Still in preliminary stages of evaluating though, so am not sure how worthwhile an extension it is; I’d be glad to share a tidied up notebook if you’re interested!
Is there a reason why you’re interested in seeing how normalising flows work with this operator in particular? I agree though that normalising flows are interesting in general.