Jobs

Courses

Resources

Companies

Placements

Community

Home >

Jobs >

Member of Technical Staff, AI - Pre-Training Platform

England, United Kingdom (On-site)

Member of Technical Staff, AI - Pre-Training Platform

1 Month ago • All levels • Research & Development • Artificial Intelligence • Undisclosed

About the job

6 skills required for this role

Add these skills to join the top 1% applicants for this job

cpp

python

data-science

cuda

scalability

game-texts

Job Description

Microsoft AI is seeking a Member of Technical Staff to contribute to its pre-training platform for advanced AI models. Responsibilities involve designing and developing Python and CUDA/HIP C++ code for distributed training of multimodal LLMs. The role includes building and maintaining infrastructure to handle petabytes of data, partnering with other teams to improve data processing, collaborating on model improvements, and identifying gaps in current model capabilities. The ideal candidate will be passionate about large-scale AI infrastructure, thrive in a collaborative environment, and have a high degree of attention to detail.

Must have:

Python and CUDA/HIP C++ development
Distributed training of multimodal LLMs
Large-scale data infrastructure experience
Collaboration and communication skills

Perks:

Industry leading healthcare
Educational resources
Discounts on products and services
Savings and investments
Maternity and paternity leave
Generous time away
Giving programs
Networking opportunities

Overview

Help build the world’s most advanced training platform at Microsoft AI

We are on a mission to create the leading pretraining platform to develop the world’s most capable AI frontier models. This platform will span one of the world’s foremost GPU clusters, pushing the boundaries of scale, performance, and reliability.

The AI Pre-training Platform team at Microsoft AI is responsible for all aspects of infrastructure including scalability, benchmarking, kernel development, performance optimizations, communications, and fault tolerance to support our model pre-training operations. We are an interdisciplinary team of engineers and scientists, learning from each other, and collaborating to create the best models, methods and products. We work closely with the teams that transform pre-trained models into the models that power the consumer Copilot experience.

We are looking for outstanding individuals excited about contributing to the next generation of systems that will transform the field. We are looking for candidates who:

Are passionate about the infrastructure enabling large-scale AI model training

Will thrive in a highly collaborative, fast-paced environment

Have a high degree of craftsmanship and pay close attention to details

Demonstrate a proactive attitude and enthusiasm for exploring new methods and technologies

Effectively manage multiple responsibilities and can adjust to shifting priorities.

Qualifications

Bachelor's Degree in Computer Science, Math, Software Engineering, Computer Engineering, or related field AND experience in business analytics, data science, software development, data modeling or data engineering work

OR Master's Degree in Computer Science, Math, Software Engineering, Computer Engineering, or related field AND experience in business analytics, data science, software development, or data engineering work

OR equivalent experience.

#Copilot #MicrosoftAI

Responsibilities

Design and develop Python and CUDA/HIP C++ code that enable distributed training of multimodal LLMs ingesting text, audio, images, or video data.
Build and maintain cutting-edge infrastructure that can store and process the petabytes of data needed to power models.
Partner with the pretraining and post-training teams to improve our data recipe by rigorous and careful experimentation.
Collaborate with the product team and other engineers and researchers across Microsoft AI to identify gaps in the current generation of models.
Embody our and .