AXLearn Apple द्वारा JAX और XLA पर आधारित एक गहन शिक्षण पुस्तकालय है, जो बड़े पैमाने पर गहन शिक्षण मॉडल विकास में सॉफ्टवेयर इंजीनियरिंग की चुनौतियों को हल करने के लिए ऑब्जेक्ट-ओरिएंटेड दृष्टिकोण का उपयोग करता है। इसकी कॉन्फ़िगरेशन प्रणाली उपयोगकर्ताओं को पुन: प्रयोज्य बिल्डिंग ब्लॉक्स से मॉडल को संयोजित करने और अन्य पुस्तकालयों (जैसे Flax और Hugging Face ट्रांसफॉर्मर्स) के साथ एकीकृत करने की अनुमति देती है। AXLearn का उद्देश्य प्रशिक्षण के पैमाने का विस्तार करना है, जिससे सैकड़ों करोड़ों मापदंडों वाले मॉडल को हजारों एक्सेलेरेटर पर कुशलतापूर्वक प्रशिक्षित किया जा सके, जो सार्वजनिक क्लाउड पर परिनियोजन के लिए उपयुक्त है। यह एक वैश्विक संगणना प्रतिमान का भी उपयोग करता है, जिससे उपयोगकर्ता व्यक्तिगत एक्सेलेरेटर के बजाय वैश्विक वर्चुअल कंप्यूटर पर संगणना का वर्णन कर सकते हैं। AXLearn प्राकृतिक भाषा प्रसंस्करण, कंप्यूटर विजन और स्पीच पहचान सहित व्यापक अनुप्रयोगों का समर्थन करता है, और इसमें अत्याधुनिक मॉडल को प्रशिक्षित करने के लिए बेसलाइन कॉन्फ़िगरेशन शामिल हैं।