Staff AI Platform Engineer: Scalable ML Infra

LinkedIn; California (MO); Hybrid; USD 175,000 - 287,000; Full time

Job summary

LinkedIn is seeking a Staff Software Engineer to scale AI training, feature engineering, and serving infrastructure across billions of parameters. You will drive high-performance data I/O, work with PyTorch and other libraries, and enable distributed training over large-scale models, while guiding containerized pipelines and observability enhancements.

The role emphasizes collaboration with open-source communities, mentorship of engineers, and shaping the future of AI platforms with a focus on

Qualifications

• Bachelor’s Degree in Computer Science or related technical discipline or equivalent practical experience.
• 4+ years of experience in the industry with leading/building deep learning systems.
• 4+ years of experience with Java, C++, Python, Go, Rust, C# and/or Functional languages such as Scala or other relevant coding languages.
• Hands-on experience developing distributed systems or other large-scale systems.

Responsibilities

• Owning the technical strategy for broad or complex requirements with insightful and forward-looking approaches that go beyond the direct team and solve large open-ended problems.
• Designing, implementing, and optimising the performance of large-scale distributed serving or training for personalised recommendation as well as large language models.
• Improving the observability and understandability of various systems with a focus on improving developer productivity and system sustenance.
• Mentoring other engineers, defining our challenging technical culture, and helping to build a fast-growing team.
• Working closely with the open-source community to participate and influence cutting-edge open-source projects.
• Functioning as the tech-lead for several concurrent key initiatives AI Infrastructure and defining the future of AI Platforms.

Skills

Distributed systems

Backend infrastructure

Java/Golang/Rust/Python

ML/DL expertise

Education

Bachelor’s degree in Computer Science or related technical discipline

Advanced degree (MS/PhD) or 4+ years relevant experience

Tools

TensorFlow

PyTorch

CUDA

cuDNN

NCCL

HuggingFace

Job description

LinkedIn is seeking a Staff Software Engineer to scale AI training, feature engineering, and serving infrastructure across billions of parameters. You will drive high-performance data I/O, work with PyTorch and other libraries, and enable distributed training over large-scale models, while guiding containerized pipelines and observability enhancements.

The role emphasizes collaboration with open-source communities, mentorship of engineers, and shaping the future of AI platforms with a focus on