Dear @Arcco96,
Let’s pool this in Tyxe/ezbnn contributions and help , and what I responded there is obviously still my answer.
Thank you for your detailed bug-reports, I fully expect things to be buggy now. Hopefully we can work on specific models once the basics of tyxe are stable and you can help us figure these bespoke models out then.