Principal AI Software Architect

Do you want to be at the forefront of innovating the latest hardware designs to propel Microsoft’s cloud growth? Are you seeking a unique career opportunity that combines technical capabilities, cross-team collaboration, with business insight and strategy? Microsoft’s mission is to empower every person and every organization on the planet to achieve more. As employees, we come together with a growth mindset, innovate to empower others, and collaborate to achieve our shared goals. Each day we build on our values of respect, integrity, and accountability to create a culture of inclusion where everyone can thrive at work and beyond. In alignment with our Microsoft values, we are committed to cultivating an inclusive work environment for all employees to positively impact our culture every day. Join the Systems Planning and Architecture (SPARC) team within Microsoft’s Azure Hardware Systems and Infrastructure (AHSI) organization, the team behind Microsoft’s expanding Cloud Infrastructure and for powering Microsoft’s "Intelligent Cloud" mission. Microsoft delivers more than 200 online services to more than one billion individuals worldwide, and AHSI is the team behind our expanding cloud infrastructure. We deliver the core infrastructure and foundational technologies for Microsoft's cloud businesses including Microsoft Azure, Bing, MSN, Office 365, OneDrive, Skype, Teams and Xbox Live. As Principal AI Software Architect for Microsoft MAIA accelerators within the SPARC team, you will play a pivotal role in enabling and optimizing machine learning model training workflows on our cutting-edge hardware. Your responsibilities will include bringing up and validating training processes on MAIA accelerators, collaborating with engineering and research teams to understand MAIA accelerator architecture, and enabling training recipes developed for Microsoft’s first-party accelerators. You will utilize your hands-on experience with PyTorch and at least one of Triton or CUDA to ensure efficient and scalable deployment of large models. This role is ideal for candidates who thrive in a dynamic environment and are passionate about advancing AI infrastructure and model performance on next-generation hardware platforms.