Staff Software Engineer, AI Platform

LinkedIn is seeking a Staff Software Engineer to join their AI Platform team, working on cutting-edge machine learning infrastructure and distributed systems. This role offers an exciting opportunity to push the boundaries of scaling large models and shape the future of AI at one of the world's largest professional networks.

The position involves working with LinkedIn's AI model training, feature engineering, and serving infrastructure, handling models with hundreds of billions of parameters. You'll be optimizing performance across various layers including algorithms, AI frameworks, data infrastructure, and hardware, while managing a GPU fleet with thousands of cards.

As a Staff Engineer, you'll lead technical strategy for complex requirements, design and implement high-performance systems, and work closely with the open-source community. The role spans multiple critical areas including Model Training Infrastructure, Feature Engineering, Model Serving Infrastructure, and MLOps.

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 with technologies like TensorFlow, PyTorch, Horovod, Ray, and various other cutting-edge AI tools.

LinkedIn offers a hybrid work environment, competitive compensation including stock options and bonuses, and comprehensive benefits. The company strongly believes in work-life balance and provides generous health and wellness programs. This is an excellent opportunity to advance your career while working on one of the most scalable AI platforms in the world.

The role can be based in Mountain View, CA, San Francisco, CA, or Bellevue, WA, offering flexibility in location while maintaining strong team collaboration. You'll be part of a talented team of researchers and engineers, with opportunities to build your personal brand in the AI industry while solving challenging technical problems at scale.