Hi Christian,
at first sight this looks exactly like what I imagined. Thanks for pointing me towards this. Unfortunately I need to finish my project first and I decided to go with the quirky work around at first. Afterwards I’m happy to try to achieve the same with nestedtensor. How would I get in touch for feedback or suggestions? Should I use this forum, a GitHub issue or something else entirely?