[6gvONxR4sf7o]

I haven't used flax, but it seems more like pytorch. I like haiku because it's relatively minimal. The simplest transform does init and that's all. I like that.

[SleekEagle]

Got it. Awesome, thanks for the info I'll check them both out this week