Senior Machine Learning Software Engineer

The SPARC organization manages Azure's hardware roadmap from architecture concept through production for all of Microsoft's current and future on-line services. This role is for a highly motivated Machine Learning Software Engineer with a strong background in neural networks, full-stack software development and hardware implementation. You will be involved with model development, low-precision data type library development, ML/HW co-design and various model optimization tools development.,

• Driving software development effort for model/HW codesign
• Developing and analysing novel NN architectures
• Implementing prototype POCs for novel low-precision data formats
• Implementing novel model architectures
• Collaborating with data scientists and ML researchers
• Interfacing with HW architecture teams
• Interfacing with SW framework teams
  
  Master's Degree/PhD in Machine learning, Computer Science, High-Performance Computing
• 4+ years of experience in ML Ops/Infra/software development in the industry
• Track record of product development and delivering commercial scale software
• Hands on experience with frameworks such as PyTorch/TensorFlow/TensorRT/ONNX RT
• Knowledge of CNN/transformer architecture and optimization strategies - quantization, sparsity, NAS, sharding, KV Cache, Flash Attention
• Strong programming skills in Python/C/C++ Experience in implementing low-level linear algebra/BLASkernels and performance optimisations
• Outstanding communication skills
  
  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.