apple/axlearn
View on GitHub ↗ ★ 2,385 Python
An Extensible Deep Learning Library
7 mentions on Hacker News by 7 people: 5 stories, 2 comments · first 2023-07-20 · latest 2025-12-17
20082026
-
It's interesting that Apple also actively develops https://github.com/apple/axlearn , which is a library on top of Jax. Seems like half the ML teams at Apple use PyTorch, and the other half uses Jax. Maybe its split between Google Cloud…
comment by ipsum2 · 2024-04-24 · in CoreNet: A library for training deep neural networks -
…was used which leverages JAX and XLA [2]. [1] https://news.ycombinator.com/item?id=39367141 [2] https://github.com/apple/axlearn
comment by refibrillator · 2024-03-16 · in MM1: Methods, Analysis and Insights from Multimodal LLM Pre-training