Taro Logo

Staff Software Engineer, AI Platform

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

Description For Staff Software Engineer, AI Platform

LinkedIn, the world's largest professional network, is seeking a Staff Software Engineer to join their AI Platform team. This is a unique opportunity to work on cutting-edge AI infrastructure that powers LinkedIn's machine learning capabilities across recommendation systems, large language models, and computer vision applications.

The role can be based in Mountain View, CA, San Francisco, CA, or Bellevue, WA, with a hybrid work arrangement that combines remote work with office presence. The team is at the forefront of scaling AI model training and serving, working with models containing hundreds of billions of parameters and managing large-scale feature engineering infrastructure.

As a Staff Software Engineer, you'll be involved in three key areas:

  1. Model Training Infrastructure: You'll build next-generation training infrastructure, optimize data I/O, collaborate with open source teams on libraries like Huggingface and PyTorch, and enable distributed training for massive models. You'll also work on containerized pipeline orchestration and maintain deep learning frameworks.

  2. Feature Engineering: You'll help develop the state-of-the-art Feature Platform that handles feature creation, computation, storage, and governance across online, offline, and nearline environments. This involves working with technologies like Spark, Beam, and Flink to process data at scale.

  3. Model Serving Infrastructure: You'll build high-performance applications for serving large and complex models, including LLMs and personalization models. This involves GPU-based inference optimization and handling significant scale requirements.

The role offers excellent growth opportunities, including working with talented researchers and engineers while building your career in the AI industry. LinkedIn provides competitive compensation, comprehensive benefits, and a culture focused on well-being and professional development.

The ideal candidate will have strong experience in distributed systems, machine learning infrastructure, and a track record of technical leadership. This is a chance to shape the future of AI infrastructure at one of the world's leading professional networks while working with cutting-edge technologies and open-source projects.

Last updated a day 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 on cutting edge projects
  • Leading key initiatives in AI Infrastructure
  • Building and optimizing model training infrastructure
  • Developing feature engineering platforms
  • Creating high-performance model serving infrastructure

Requirements For Staff Software Engineer, AI Platform

Python
Java
Go
Rust
Scala
Kubernetes
  • 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
  • Hands-on experience developing distributed systems or other large-scale systems
  • Outstanding interpersonal communication skills
  • Experience with machine learning infrastructure
  • Experience with distributed data processing engines

Benefits For Staff Software Engineer, AI Platform

Medical Insurance
401k
Parental Leave
  • Generous health and wellness programs
  • Time away for employees of all levels
  • Annual performance bonus
  • Stock options
  • Comprehensive benefits package

Interested in this job?

Jobs Related To LinkedIn Staff Software Engineer, AI Platform

Senior Staff Engineer, Machine Learning - Notifications AI

Senior Staff Engineer position at LinkedIn leading the Notifications AI team, focusing on machine learning and personalization systems with competitive compensation $191K-$315K.

Sr Staff Engineer, ML Infrastructure and Performance

Senior Staff Engineer role at LinkedIn focusing on ML infrastructure, GPU computing, and performance optimization. Hybrid role in Sunnyvale, CA with competitive compensation.

Senior Staff AI Engineer, AI Algorithm Foundations

Senior Staff AI Engineer position at LinkedIn focusing on developing next-generation AI algorithms and foundational models for professional networking platform serving over billion members.

Senior Staff Engineer, Machine Learning - Notifications AI

Senior Staff Engineer position at LinkedIn focusing on Machine Learning and AI for notifications systems, offering competitive compensation and the opportunity to work with cutting-edge AI technologies.

Senior Staff AI Engineer, AI Algorithm Foundations

Lead AI engineer role at LinkedIn focusing on developing next-gen recommender systems and LLMs, offering competitive compensation and opportunity to shape the future of professional networking through AI innovation.