apple/axlearn
An Apple-developed extensible deep learning library built on JAX and XLA for training and scaling large models.
Not currently ranked — collecting fresh signals.
star history
AXLearn is an open-source library maintained by Apple that provides infrastructure for developing and training large-scale deep learning models. It is built on top of JAX and XLA, leveraging their auto-vectorization and JIT compilation capabilities for efficient computation on CPU, GPU, and TPU hardware. The library emphasizes an object-oriented approach to model configuration and includes CLI tooling for training workflows.
Frequently asked
- What is apple/axlearn?
- An Apple-developed extensible deep learning library built on JAX and XLA for training and scaling large models.
- Is axlearn open source?
- Yes — apple/axlearn is open source, released under the Apache-2.0 license.
- What language is axlearn written in?
- apple/axlearn is primarily written in Python.
- How popular is axlearn?
- apple/axlearn has 2.4k stars on GitHub.
- Where can I find axlearn?
- apple/axlearn is on GitHub at https://github.com/apple/axlearn.