Hacker News new | ask | show | jobs
by p1esk 642 days ago
I wish Jax had everything I need to experiment with DL models built in, natively - like Pytorch. Instead there are many third party libraries (flax, trax, haiku, this one, etc). I have no idea which one to use. This was the case when I first played with jax 5 years ago, and it’s still the case today (even worse it seems). This makes it a non starter for me.
1 comments
__rito__ 641 days ago
Why?

Use any. I used to work with Flax, now I work with Equinox more. Choose any between Flax, Equinox, and Haiku.

link
p1esk 641 days ago
Use any

Too much overhead when I just want to get shit done.

link
__rito__ 640 days ago
Okay, I am telling you to use Flax.

Its high level API is quite similar to that of PyTorch. So, you will feel right at home.

link
p1esk 640 days ago
So, “do as I say, not as I do”? I just looked at Equinox and it does look a little better than flax, but flax seems to be more widely used. Both haiku and trax seem to be on the way out. All four were intended to do exactly the same thing. What a mess of an ecosystem.
link