apple/axlearn
An Extensible Deep Learning Library
This library helps machine learning engineers and researchers build, train, and manage very large-scale deep learning models for applications like natural language processing, computer vision, and speech recognition. It takes deep learning model architectures and data as input, and outputs trained models ready for deployment or further research. This is for professionals working with massive datasets and models requiring thousands of accelerators.
2,331 stars.
Use this if you are a deep learning engineer or researcher tasked with developing and training state-of-the-art models with billions of parameters across a distributed system of thousands of accelerators.
Not ideal if you are working on smaller-scale deep learning projects that don't require distributed training across numerous accelerators or if you prefer a simpler, less abstract framework.
Stars
2,331
Forks
403
Language
Python
License
Apache-2.0
Category
Last pushed
Feb 18, 2026
Commits (30d)
0
Get this data via API
curl "https://pt-edge.onrender.com/api/v1/quality/ml-frameworks/apple/axlearn"
Open to everyone — 100 requests/day, no key needed. Get a free key for 1,000/day.
Related frameworks
explosion/thinc
🔮 A refreshing functional take on deep learning, compatible with your favorite libraries
google-deepmind/optax
Optax is a gradient processing and optimization library for JAX.
patrick-kidger/diffrax
Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capable....
google/grain
Library for reading and processing ML training data.
patrick-kidger/equinox
Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/