Staff Software Engineer, AI Platform

LinkedIn is seeking a Staff Software Engineer for their AI Platform team to push the boundaries of scaling large models. This role offers an exciting opportunity to work on cutting-edge AI infrastructure at one of the world's largest professional networks.

The position involves working with state-of-the-art AI technologies, including large language models, recommendation systems, and computer vision models. You'll be responsible for scaling LinkedIn's AI model training, feature engineering, and serving infrastructure handling models with hundreds of billions of parameters. The role spans multiple technical areas including AI frameworks, data infrastructure, and hardware optimization for GPU fleets.

The team has a strong open-source presence, with many team members being committers to prominent projects like TensorFlow, Horovod, Ray, vLLM, and DeepSpeed. You'll have the opportunity to contribute to these communities while building LinkedIn's next-generation AI infrastructure.

Key focus areas include:

• Model Training Infrastructure: Building high-performance data I/O systems and enabling distributed training for massive models
• Feature Engineering: Developing platforms for feature creation, computation, storage, and governance across online/offline environments
• Model Serving Infrastructure: Creating low-latency applications for serving large complex models
• MLOps: Managing infrastructure for ML operations and experimentation systems

The role offers competitive compensation ($156,000 - $255,000) and benefits, including comprehensive health coverage and wellness programs. LinkedIn's hybrid work environment balances remote work with in-office collaboration, fostering a culture of trust and flexibility.

This is an excellent opportunity for experienced engineers passionate about AI infrastructure to make a significant impact at scale. You'll work with talented researchers and engineers while building your career and personal brand in the AI industry. The position requires strong technical expertise combined with leadership skills, as you'll be guiding technical strategy and mentoring other engineers.