Software Engineer, AI Platform

LinkedIn is the world's largest professional network, dedicated to creating economic opportunities for its members. The Software Engineer role within the AI Platform team involves designing and optimizing large-scale distributed systems for AI model training and serving, while collaborating with various teams to enhance performance and feature engineering.

Responsibilities

• Designing, implementing, and optimizing the performance of large-scale distributed serving or training for personalized recommendation as well as large language models
• Improving the observability and understandability of various systems with a focus on improving developer productivity and system sustenance
• Partner with peers, leads and partners to define, scope, prioritize, and build impactful features at a high velocity

Skills

• Bachelor's Degree in Computer Science or related technical discipline, or equivalent practical experience
• 1+ years of experience in the industry with leading/ building deep learning systems
• Experience with Java, C++, Python, Go, Rust, C# and/or Functional languages such as Scala or other relevant coding languages
• Experience, qualifications in Machine Learning, AI
• 2+ years of relevant work experience
• MS or PhD in Computer Science or related technical discipline
• Experience building ML applications, LLM serving, GPU serving
• Experience with search systems or similar large-scale distributed systems
• Experience with distributed data processing engines like Flink, Beam, Spark etc., feature engineering
• Experience in distributed machine learning training infrastructure, including technologies like Horovod, PyTorch FSDP, DeepSpeed, Hugginface, PyTorch Lightning, LLMs, GNNs, MLFlow, Kubeflow and large scale distributed systems
• Familiarity with containers and container orchestration systems like Kubernetes
• Experience in deep learning frameworks and tensor libraries like PyTorch, Tensorflow, JAX/FLAX

Benefits

• Generous health and wellness programs
• Annual performance bonus
• Stock
• Benefits
• Other applicable incentive compensation plans

Company Overview

LinkedIn is a professional networking site that allows users to create business connections, search for jobs, and find potential clients. It is a sub-organization of Microsoft. It was founded in 2003, and is headquartered in Sunnyvale, California, USA, with a workforce of 10001+ employees. Its website is https://www.linkedin.com.

Company H1B Sponsorship

LinkedIn has a track record of offering H1B sponsorships, with 892 in 2025, 1108 in 2024, 913 in 2023, 1580 in 2022, 1043 in 2021, 1146 in 2020. Please note that this does not guarantee sponsorship for this specific role.

  Apply Now