Daniel Ward

It could be possible to support a transformer with shape/dimension matching the shape of the total bijection (rather than only scalar bijections), in which case you could ``Stack``/``Concatenate`` transformers as...

Thanks for the suggestion and the motivation to start thinking about this. It would improve the robustness and user friendliness of the package. One example that I've fallen into a...

I've started adding it here https://github.com/danielward27/flowjax/pull/154. The tests now pass whilst running beartype. I've added some shape annotations, although I'm sure more detail could be added in places. I'll probably...

I'll close this for now, runtime type checking is great, but the options are not massively mature (also pythons typing system is a bit of a mess at the best...

That would definitely be interesting to have