Yes. I think the point of equinox (or nnx) API is that a model class can manage both function and data, which makes my simple brain happy. But if you are OK to maintain a dict of parameters anywhere in the code, linen (or haiku) is just OK.
Check my blog kngwyu.github.io/rlog2/posts/... for more

I'm a long-time user of equinox, which is very similar to nnx.
I'm not sure why they re-invented nnx instead of just copy-pasting equinox to their library... 😔

Thank you for your suggestion! But since I need to make my publication list a bit longer for my postdoc grant proposal, I would not have enough time to work for short papers 😓