I am also interested in this, and I believe many others would be as well. I list some packages that target this particular problem below for reference
- https://github.com/ahwillia/CatViews.jl
- GitHub - rafaqz/Flatten.jl: Flatten nested Julia objects to tuples, and reconstruct them later (Discourse ann.)
- https://github.com/rafaqz/Nested.jl
- https://github.com/simonbyrne/PackVec.jl
I might attempt at some implementation of packing/unpacking the Zygote.Grads type, which perhaps would be well received in the Flux repository.
If you made any strides forward on this problem I would be interested in learning from your experience. I will post here once I have a working solution.