Re:Jax and Equinox: What are they and why should I bother?
by spearman on Friday September 13, 2024 @12:26AM
(#41528019)
Pretty sure it's Apple building it, and they're using JAX in-house so I imagine it will get better over time. Though they do love to drop support for old things so maybe M1 will never work again...
[ Parent
| on HN ]
Re:Jax and Equinox: What are they and why should I bother?
by mccoyb on Friday September 13, 2024 @06:28AM
(#41529825)
I think they’re likely using MLX in house now, no? (Probably not everyone, ofc - but seems likely that many will just use the native array framework designed explicitly for MX chips)
Re:Jax and Equinox: What are they and why should I bother?
by spearman on Friday September 13, 2024 @04:25PM
(#41534790)
From https://machinelearning.apple.com/research/introducing-apple...
> Our foundation models are trained on Apple's AXLearn framework, an open-source project we released in 2023. It builds on top of JAX