Wednesday, October 2, 2024

Azure Maia for the period of AI: From silicon to software program to programs

Because the tempo of AI and the transformation it permits throughout industries continues to speed up, Microsoft is dedicated to constructing and enhancing our international cloud infrastructure to fulfill the wants from prospects and builders with quicker, extra performant, and extra environment friendly compute and AI options. Azure AI infrastructure includes know-how from {industry} leaders in addition to Microsoft’s personal improvements, together with Azure Maia 100, Microsoft’s first in-house AI accelerator, introduced in November. On this weblog, we’ll dive deeper into the know-how and journey of growing Azure Maia 100, the co-design of {hardware} and software program from the bottom up, constructed to run cloud-based AI workloads and optimized for Azure AI infrastructure.

Azure Maia 100, pushing the boundaries of semiconductor innovation

Maia 100 was designed to run cloud-based AI workloads, and the design of the chip was knowledgeable by Microsoft’s expertise in operating complicated and large-scale AI workloads similar to Microsoft Copilot. Maia 100 is without doubt one of the largest processors made on 5nm node utilizing superior packaging know-how from TSMC.   

Via collaboration with Azure prospects and leaders within the semiconductor ecosystem, similar to foundry and EDA companions, we’ll proceed to use real-world workload necessities to our silicon design, optimizing your complete stack from silicon to service, and delivering the most effective know-how to our prospects to empower them to realize extra.

Azure Maia 100, Microsoft’s first in-house AI accelerator

Finish-to-end programs optimization, designed for scalability and sustainability 

When growing the structure for the Azure Maia AI accelerator sequence, Microsoft reimagined the end-to-end stack in order that our programs may deal with frontier fashions extra effectively and in much less time. AI workloads demand infrastructure that’s dramatically totally different from different cloud compute workloads, requiring elevated energy, cooling, and networking functionality. Maia 100’s {custom} rack-level energy distribution and administration integrates with Azure infrastructure to realize dynamic energy optimization. Maia 100 servers are designed with a fully-custom, Ethernet-based community protocol with mixture bandwidth of 4.8 terabits per accelerator to allow higher scaling and end-to-end workload efficiency.  

After we developed Maia 100, we additionally constructed a devoted “sidekick” to match the thermal profile of the chip and added rack-level, closed-loop liquid cooling to Maia 100 accelerators and their host CPUs to realize larger effectivity. This structure permits us to convey Maia 100 programs into our current datacenter infrastructure, and to suit extra servers into these services, all inside our current footprint. The Maia 100 sidekicks are additionally constructed and manufactured to fulfill our zero waste dedication. 

The server rack and cooling “sidekick” for Azure Maia 100

Co-optimizing {hardware} and software program from the bottom up with the open-source ecosystem 

From the beginning, transparency and collaborative development have been core tenets in our design philosophy as we construct and develop Microsoft’s cloud infrastructure for compute and AI. Collaboration permits quicker iterative improvement throughout the {industry}—and on the Maia 100 platform, we’ve cultivated an open neighborhood mindset from algorithmic knowledge varieties to software program to {hardware}.  

To make it straightforward to develop AI fashions on Azure AI infrastructure, Microsoft is creating the software program for Maia 100 that integrates with well-liked open-source frameworks like PyTorch and ONNX Runtime. The software program stack offers wealthy and complete libraries, compilers, and instruments to equip knowledge scientists and builders to efficiently run their fashions on Maia 100. 

Diagram showing the software stack of Azure Maia 100

To optimize workload efficiency, AI {hardware} sometimes requires improvement of {custom} kernels which are silicon-specific. We envision seamless interoperability amongst AI accelerators in Azure, so we’ve built-in Triton from OpenAI. Triton is an open-source programming language that simplifies kernel authoring by abstracting the underlying {hardware}. It will empower builders with full portability and suppleness with out sacrificing effectivity and the flexibility to focus on AI workloads. 

Diagram showing the seamless development experience on Azure AI infrastructure

Maia 100 can also be the primary implementation of the Microscaling (MX) knowledge format, an industry-standardized knowledge format that results in quicker mannequin coaching and inferencing instances. Microsoft has partnered with AMD, ARM, Intel, Meta, NVIDIA, and Qualcomm to launch the v1.0 MX specification by the Open Compute Mission neighborhood in order that your complete AI ecosystem can profit from these algorithmic enhancements. 

Azure Maia 100 is a novel innovation combining state-of-the-art silicon packaging methods, ultra-high-bandwidth networking design, fashionable cooling and energy administration, and algorithmic co-design of {hardware} with software program. We look ahead to persevering with to advance our objective of creating AI actual by introducing extra silicon, programs, and software program improvements into our datacenters globally.

Be taught extra 



Related Articles

LEAVE A REPLY

Please enter your comment!
Please enter your name here

Latest Articles