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

Company H1B Sponsorship