Machine Learning Engineer / Machine Learning Engineeress

Overview

The Spatial AI Lab is an integral part of the Applied Sciences Group at Microsoft, a pioneering research and development organization dedicated to crafting next-generation human-computer interaction technologies. We leverage cutting-edge advancements in artificial intelligence while exploring innovative hardware capabilities and device form factors. Our talented team of scientists and engineers specializes in computer vision, multi-modal AI, and spatial and embodied AI.

As a Machine Learning Engineer, you will collaborate closely with various research and product teams, contributing to the development of compelling new experiences in the market. Many of these experiences will be driven by computer vision and multimodal AI models. Your role will involve data collection, model evaluation and training, as well as writing production-quality code. Additionally, you will have the opportunity to participate in groundbreaking research alongside esteemed partners like ETH Zurich, publish in top-tier venues, present at workshops, and mentor students.

At Microsoft, our mission is to empower every individual and organization on the planet to achieve more. We foster a culture of growth, innovation, and collaboration, where employees unite to realize shared goals. Each day, we build upon our values of respect, integrity, and accountability, creating an inclusive environment where everyone can thrive both at work and beyond.

Responsibilities

• Implement algorithms, design model architectures, run experiments, perform evaluations, and build data pipelines.
• Develop data and learning solutions ensuring scalability, efficiency, and performance.
• Curate training and evaluation datasets/benchmarks.
• Optimize deep neural networks for deployment on Neural Processing Units (NPUs), GPUs, and cloud environments, maximizing efficiency and performance.

Qualifications

Required Qualifications:

• Master’s degree in Computer Science or 3+ years of relevant industry experience.
• Proficiency in programming languages such as Python and/or C++.
• Hands-on experience with modern deep learning frameworks (e.g., PyTorch, TensorFlow, JAX).
• Self-motivated team player with strong problem-solving skills and a desire to learn.

Preferred Qualifications:

• Ability to present complex technical concepts to diverse audiences.
• Experience quantizing deep neural networks for NPUs and GPUs, and familiarity with the ONNX format.
• Hands-on experience in training and fine-tuning deep neural networks.
• Knowledge of distributed computing frameworks (e.g., Ray) and experience using Azure Machine Learning (AzureML) for model training and deployment.
• Background in building agent-based systems.

This position is open for a minimum of 5 days, with applications accepted on an ongoing basis until filled. Apply online using the form below. Only applications matching the job profile will be considered.

Microsoft is an equal opportunity employer. All qualified applicants will receive consideration for employment without regard to age, ancestry, citizenship, color, family or medical care leave, gender identity or expression, genetic information, immigration status, marital status, medical condition, national origin, physical or mental disability, political affiliation, protected veteran or military status, race, ethnicity, religion, sex (including pregnancy), sexual orientation, or any other characteristic protected by applicable local laws, regulations, and ordinances.