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.