Thanks for answering! Sure, I’m using Lux now too. Still, a lot of people use Flux. I’ll just open an issue in Flux and see what happens over there. Flux issue: Restructure for ComponentArrays.jl unreasonably slow · Issue #2238 · FluxML/Flux.jl · GitHub