Staff Software Engineer, AI Platform

LinkedIn is seeking a Staff Software Engineer to join their AI Platform team, focusing on pushing the boundaries of scaling large models. This role presents an exciting opportunity to work at the intersection of AI infrastructure and large-scale systems.

The position involves working with cutting-edge technologies in AI model training, feature engineering, and serving infrastructure. You'll be responsible for scaling LinkedIn's AI capabilities, handling models with hundreds of billions of parameters, and optimizing performance across various layers of the tech stack.

The role is split across several key areas:

• Model Training Infrastructure: Building next-gen training infrastructure for AI use cases, implementing high-performance data I/O, and working with popular libraries like Huggingface, Horovod, and PyTorch
• Feature Engineering: Developing state-of-the-art Feature Platform for AI users, working with technologies like Spark, Beam, and Flink
• Model Serving Infrastructure: Creating low-latency applications for serving large & complex models, including LLMs and Personalization models
• ML Ops: Managing infrastructure for MLOps and experimentation systems

This is a hybrid position available in Mountain View, CA, San Francisco, CA, or Bellevue, WA. The role offers competitive compensation ($156,000 - $255,000) and the opportunity to work with talented teams of researchers and engineers while building your career in the AI industry.

The ideal candidate will have 8+ years of relevant experience, strong expertise in distributed systems, and deep knowledge of machine learning infrastructure. You'll be working in a collaborative environment, contributing to open-source projects, and helping shape the future of AI at LinkedIn.

This role is perfect for someone who wants to make a significant impact on AI infrastructure at scale while working with a world-class team in a company that values innovation and professional growth.