Senior Machine Learning Research Engineer

1 Month ago • 4-8 Years • DevOps

Job Summary

Job Description

This Senior Machine Learning Research Engineer role at Microsoft's Azure Hardware Systems & Infrastructure (AHSI) involves driving model/hardware co-design, developing novel neural network architectures and low-precision data formats, and collaborating with data scientists and hardware/software teams. Responsibilities include analyzing NN architectures, inventing model architectures and low-precision data formats, and optimizing performance. The ideal candidate possesses expertise in ML systems, model optimization, efficient model architectures (CNNs, transformers), and low-level linear algebra. Experience with PyTorch/TensorFlow/TensorRT and strong programming skills in Python/C/C++ are essential.
Must have:
  • MS/PhD in relevant field
  • Extensive ML systems experience
  • Model optimization expertise
  • CNN/Transformer architecture knowledge
  • Python/C/C++ programming skills
Good to have:
  • Experience with quantization, sparsity, NAS
  • Low-level linear algebra/BLAS kernel implementation
  • Experience with KV Cache, Flash Attention
Perks:
  • Industry-leading healthcare
  • Educational resources
  • Product & service discounts
  • Savings and investment plans
  • Maternity/paternity leave
  • Generous time off
  • Giving programs
  • Networking opportunities

Job Details

Overview

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 both technical capabilities, cross team collaboration, with business insight and strategy?  

 

Join our Strategic Planning and Architecture (SPARC) team within Microsoft’s Azure Hardware Systems & Infrastructure (AHSI) organization and be a part of the organization behind Microsoft’s expanding Cloud Infrastructure and responsible 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.   

 

The SPARC organization manages Azure’s hardware roadmap from architecture concept through production for all of Microsoft’s current and future on-line servicesThis role is for a highly motivated Machine Learning Engineer with a strong background in neural networks and hardware implementation. You will be involved with both model development, data type analysis, ML/HW co-design. 

Qualifications

  • MS/PhD in Machine learning, Computer Architecture/Systems, High-Performance Computing or related areas 
  • Extensive  experience in ML systems/Model optimizations/Efficient model architecture 
  • Track record of original research and delivering novel results in ML systems area 
  • Hands on experience with frameworks such as PyTorch/TensorFlow/TensorRT 
  • Deep 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/BLAS kernels and performance optimisations 
  • Outstanding communication skills 

 

Responsibilities

 

  • Driving model/HW codesign 
  • Developing and analysing novel NN architectures 
  • Inventing novel low-precision data formats 
  • Inventing novel model architectures 
  • Collaborating with data scientists and ML researchers 
  • Interfacing with HW architecture teams 
  • Interfacing with SW framework teams 
Benefits/perks listed below may vary depending on the nature of your employment with Microsoft and the country where you work.
Industry leading healthcare
Educational resources
Discounts on products and services
Savings and investments
Maternity and paternity leave
Generous time away
Giving programs
Opportunities to network and connect

Similar Jobs

Luxoft - Senior GPU Compiler Software Development Engineer

Luxoft

Poland, Ohio, United States (Remote)
3 Months ago
N-iX - ML Engineer (Computer Vision) (#2424)

N-iX

Colombia (Remote)
3 Months ago
Ello - Tech Lead, Machine Learning

Ello

Canada (On-Site)
3 Months ago
Intel Corporation - CCG CGAI: AI Software Solutions Engineer

Intel Corporation

(On-Site)
2 Months ago
Salesforce - Serener Engineering Manager-Distributed Systems

Salesforce

Hyderabad, Telangana, India (Hybrid)
3 Months ago
Admin Looks - Release Manager

Admin Looks

Hyderabad, Telangana, India (Remote)
3 Months ago
Luxoft - JavaScript Full Stack Engineer

Luxoft

Kuala Lumpur, Federal Territory Of Kuala Lumpur, Malaysia (On-Site)
3 Months ago
Nagarro - Senior Engineer, DevOps

Nagarro

Mumbai, Maharashtra, India (On-Site)
4 Months ago

Get notifed when new similar jobs are uploaded

Similar Skill Jobs

Passive Logic - AI Control Theory & Optimization Scientist

Passive Logic

Salt Lake City, Utah, United States (On-Site)
2 Months ago
G5 Games - 2D UI/UX Artist (match-3 project)

G5 Games

Yerevan, Yerevan, Armenia (Remote)
4 Months ago
Playrix - Game Designer

Playrix

Ireland (Remote)
4 Months ago
Arkose Labs - Senior Machine Learning Researcher

Arkose Labs

Pune, Maharashtra, India (Hybrid)
4 Months ago
G5 Games - 2D Illustrator (HOG project)

G5 Games

Yerevan, Yerevan, Armenia (Remote)
2 Months ago
Luxoft - Senior PySpark Data Engineer

Luxoft

(Remote)
3 Months ago
G5 Games - 2D Illustrator (Match-3 project)

G5 Games

Yerevan, Yerevan, Armenia (Remote)
2 Months ago
Ascent Health - Lead Data Scientist

Ascent Health

Karnataka, India (On-Site)
3 Months ago
Microsoft - Research Intern - Applied Sciences Group (Computer Agent)

Microsoft

Redmond, Washington, United States (On-Site)
1 Month ago
Playrix - Feature Owner (LiveOps)

Playrix

Montenegro (Remote)
4 Months ago

Get notifed when new similar jobs are uploaded

Jobs in Cambridge, England, United Kingdom

Meta - Software Engineer (Leadership) - Infrastructure

Meta

London, England, United Kingdom (On-Site)
3 Months ago
Sumo Digital - Technical Designer

Sumo Digital

Leeds, England, United Kingdom (On-Site)
3 Months ago
WebMD - Data Analyst (m/w/d)

WebMD

United Kingdom (On-Site)
4 Months ago
Gunzilla - Art Production Director

Gunzilla

London, England, United Kingdom (On-Site)
1 Month ago
Microsoft - Member of Technical Staff, AI Data

Microsoft

London, England, United Kingdom (On-Site)
1 Month ago
Media Molecule - Lead UX / UI Programmer

Media Molecule

London, England, United Kingdom (Hybrid)
8 Months ago
Aristocrat Gaming - Development Manager

Aristocrat Gaming

London, England, United Kingdom (Hybrid)
1 Month ago
Twitch - Senior Manager, EMEA Partnerships

Twitch

London, England, United Kingdom (On-Site)
1 Month ago
Rockstar Games - Animation R&D Programmer (Mid / Senior Level)

Rockstar Games

Edinburgh, Scotland, United Kingdom (On-Site)
4 Months ago
Assystems - Manufacturing Engineer

Assystems

Derby, England, United Kingdom (On-Site)
3 Months ago

Get notifed when new similar jobs are uploaded

DevOps Jobs

Alcon - DevOps Engineer

Alcon

Bengaluru, Karnataka, India (Hybrid)
4 Months ago
HID - Lead DevOps Engineer

HID

Chennai, Tamil Nadu, India (Hybrid)
3 Months ago
Glean - SRE Manager (India)

Glean

Bengaluru, Karnataka, India (On-Site)
3 Months ago
Luxoft - Azure Enterprise Architect

Luxoft

Warsaw, Masovian Voivodeship, Poland (On-Site)
3 Months ago
Canva - Senior Software Engineer - Cloud Security & Compliance, remote across ANZ

Canva

Sydney, New South Wales, Australia (Remote)
2 Months ago
InvenioLSI - MuleSoft Architecht

InvenioLSI

Dubai, Dubai, United Arab Emirates (On-Site)
2 Months ago
Luxoft - Senior Application Engineer

Luxoft

Kraków, Lesser Poland Voivodeship, Poland (On-Site)
3 Months ago
PwC - Azure Cloud Solutions Architect, Senior Manager

PwC

Toronto, Ontario, Canada (On-Site)
2 Months ago
Microsoft - Critical Environment Technical Trainer

Microsoft

Jakarta, Jakarta, Indonesia (On-Site)
1 Month ago

Get notifed when new similar jobs are uploaded

About The Company

Microsoft is a tech giant that develops, licenses, and supports a range of software products, services, and devices.

London, England, United Kingdom (On-Site)

Mountain View, California, United States (Hybrid)

Mountain View, California, United States (Hybrid)

Mountain View, California, United States (Hybrid)

New York, New York, United States (Hybrid)

Mountain View, California, United States (Hybrid)

Mountain View, California, United States (Hybrid)

London, England, United Kingdom (On-Site)

Dublin, County Dublin, Ireland (On-Site)

View All Jobs

Get notified when new jobs are added by Microsoft

Level Up Your Career in Game Development!

Transform Your Passion into Profession with Our Comprehensive Courses for Aspiring Game Developers.

Job Common Plug