Yuji Kanagawa
kngwyu.bsky.social
Yuji Kanagawa
@kngwyu.bsky.social
Okinawa, Japan
ALIFE, neuroscience, running, and some random photos
https://kngwyu.github.io/
October 3, 2025 at 10:41 AM
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
Try reinforcement learning with equinox – RLog2
kngwyu.github.io
August 10, 2025 at 12:37 PM
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... 😔
August 10, 2025 at 6:40 AM
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 😓
August 4, 2025 at 6:00 AM