Increased AD compilation times induced by the change in #67

[simsurace]

While the compiled function runs much faster due to type instability, for bigger tuples this leads to a blowup of compilation time for Zygote when differentiating through the unflatten function defined recursively (unsurprisingly perhaps). We should maybe add an rrule.