> Our foundation models are trained on Apple's AXLearn framework, an open-source project we released in 2023. It builds on top of JAX