Senior Machine Learning Research Scientist
Cambridge, Cambridgeshire, United Kingdom
Job number: 1791677
Work site: Up to 100% work from home
Travel: None
Role type: Research, Applied, & Data Sciences
Employment type: Full-Time
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.
The SPARC organization manages Azure’s hardware roadmap from architecture concept through production for all of Microsoft’s current and future online services. This 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, and ML/HW co-design.
Qualifications
* Master's Degree/PhD in Machine Learning, Computer Architecture/Systems, High-Performance Computing or related areas.
* 3+ years of experience in ML systems, model optimizations, and 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, or TensorRT.
* Deep knowledge of CNN and transformer architecture and optimization strategies – quantization, sparsity, NAS, sharding, KV Cache, Flash Attention.
* Strong programming skills in Python, C, or C++.
* Experience in implementing low-level linear algebra/BLAS kernels and performance optimizations.
* Knowledge of GPU, TPU, or similar NPU accelerator architecture.
Responsibilities
* Driving model/hardware co-design.
* Developing and analyzing novel LLM architectures.
* Inventing novel low-precision data/number formats for training/inference of SOTA LLMs.
* Collaborating with data scientists and ML researchers.
* Interfacing with HW architecture teams.
* Interfacing with SW framework teams.
Benefits
* 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
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. If you need assistance and/or a reasonable accommodation due to a disability during the application process, read more about requesting accommodations.
#J-18808-Ljbffr