←back to thread

Jax and Equinox: What are they and why should I bother?

(www.garymm.org)
58 points spearman | 2 comments | 09 Sep 24 14:46 UTC | HN request time: 0.697s | source
Show context
p1esk ◴[13 Sep 24 03:39 UTC] No.41527857[source]▶
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.
replies(1): >>41534589 #
__rito__ ◴[13 Sep 24 19:59 UTC] No.41534589[source]▶
Why?

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

replies(1): >>41535941 #
p1esk ◴[13 Sep 24 22:54 UTC] No.41535941[source]▶
Use any

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

replies(1): >>41546110 #
1. __rito__ ◴[15 Sep 24 08:25 UTC] No.41546110[source]▶
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.

replies(1): >>41546862 #
2. p1esk ◴[15 Sep 24 11:39 UTC] No.41546862[source]▶
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.