Staff Software Engineer, AI Platform

LinkedIn is the world's largest professional network, built to help members achieve more in their careers.
$156,000 - $255,000
Machine Learning
Staff Software Engineer
Hybrid
5,000+ Employees
4+ years of experience
AI · Enterprise SaaS

Description For Staff Software Engineer, AI Platform

LinkedIn is seeking a Staff Software Engineer to join their AI Platform team, focusing on scaling large model training and serving infrastructure. The role involves pushing the boundaries of AI scaling, working with models containing hundreds of billions of parameters across recommendation systems, large language models, and computer vision applications. The team optimizes performance across algorithms, frameworks, data infrastructure, and hardware to maximize their GPU fleet's capabilities.

The position spans several key areas including Model Training Infrastructure, Feature Engineering, Model Serving Infrastructure, and MLOps. You'll be working with cutting-edge technologies and frameworks like TensorFlow, Horovod, Ray, vLLM, Huggingface, and DeepSpeed. The team actively contributes to the open source community and includes many open source committers.

As a Staff Engineer, you'll lead technical strategy development, implement large-scale distributed systems, improve system observability, and mentor other engineers. You'll work on challenging problems like scaling models with hundreds of billions of parameters, optimizing training and serving infrastructure, and building robust feature engineering platforms handling millions of QPS.

The role offers the opportunity to shape the future of AI infrastructure at one of the world's largest professional networks. You'll collaborate with talented researchers and engineers while building your career and personal brand in the AI industry. LinkedIn offers a hybrid work environment, combining remote work with in-office collaboration on select days.

The position requires strong technical leadership, deep expertise in distributed systems and AI infrastructure, and the ability to mentor others while driving technical excellence. You'll be at the forefront of solving complex challenges in AI scaling while working with a world-class team in a collaborative environment.

Last updated 3 days ago

Responsibilities For Staff Software Engineer, AI Platform

  • Owning the technical strategy for broad or complex requirements
  • Designing and implementing large-scale distributed serving or training for personalized recommendation and language models
  • Improving system observability and developer productivity
  • Mentoring other engineers and defining technical culture
  • Working with open-source community
  • Functioning as tech-lead for key AI Infrastructure initiatives

Requirements For Staff Software Engineer, AI Platform

Python
Java
Go
Rust
Scala
  • Bachelor's Degree in Computer Science or related technical discipline
  • 4+ years of experience in the industry with leading/building deep learning systems
  • 4+ years of experience with Java, C++, Python, Go, Rust, C# or Functional languages
  • Hands-on experience developing distributed systems

Benefits For Staff Software Engineer, AI Platform

Medical Insurance
401k
Parental Leave
  • Hybrid work environment
  • Career growth opportunities
  • Work with cutting-edge AI technologies

Interested in this job?

Jobs Related To LinkedIn Staff Software Engineer, AI Platform

Sr. Staff Software Engineer, AI Infra

Senior Staff Software Engineer position at LinkedIn focusing on AI infrastructure, distributed systems, and large-scale model training.

AI Engineering Manager, Enterprise AI

AI Engineering Manager position at LinkedIn leading Enterprise AI initiatives across Hirer, Learning and Enterprise Jobs verticals, managing a team of 6-10 engineers.

AI Engineering Manager, Enterprise AI

Lead AI engineering team at LinkedIn developing enterprise AI solutions for recruiting, learning, and job matching platforms.

AI Engineering Manager, Trust / Anti-Abuse AI

Lead LinkedIn's Anti-Abuse AI team developing next-gen modeling systems to combat fake accounts and harmful activities.

AI Engineering Manager, Trust / Anti-Abuse AI

Lead LinkedIn's Anti-Abuse AI team developing next-gen modeling to combat fake accounts and harmful activities while ensuring platform safety and authenticity.