How to implement "composite" layers?

For example in #208 @OneAdder describes a layer that does multiple chained operations associated with simpler layers. Transformers are similar. How to best designed a unified API for composite layers?