Conditional Coupling Layers

[rudolfwilliam]

It seems that some layers do not support conditioning, like the affine coupling block. Would it be possible to extend this, as suggested in e.g. https://arxiv.org/pdf/1912.00042 ? Wouldn't be much effort, just need to make the affine coupling conditional and the rest can stay.