Software Engineer, AI Platform jobs in United States
cer-icon
Apply on Employer Site
company-logo

LinkedIn · 4 hours ago

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.

Professional NetworkingRecruitingSocial MediaSocial Recruiting
check
Comp. & Benefits
check
H1B Sponsor Likelynote

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

Qualification

Deep Learning SystemsMachine LearningGPU ServingDistributed SystemsPythonJavaC++TensorFlowPyTorchKubernetesSoft Skills

Required

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

Preferred

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 and/or other applicable incentive compensation plans

Company

LinkedIn

company-logo
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.

H1B Sponsorship

LinkedIn has a track record of offering H1B sponsorships. Please note that this does not guarantee sponsorship for this specific role. Below presents additional info for your reference. (Data Powered by US Department of Labor)
Distribution of Different Job Fields Receiving Sponsorship
Represents job field similar to this job
Trends of Total Sponsorships
2025 (892)
2024 (1108)
2023 (913)
2022 (1580)
2021 (1043)
2020 (1146)

Funding

Current Stage
Public Company
Total Funding
$154.8M
Key Investors
Q4 Investments (Pty) Ltd,TCVBain Capital VenturesBessemer Venture Partners,Global Founders Capital
2016-06-13Acquired
2016-02-15Private Equity
2014-04-01Series Unknown

Leadership Team

leader-logo
Ryan Roslansky
Chief Executive Officer
linkedin
leader-logo
Dan Shapero
Chief Operating Officer
linkedin
Company data provided by crunchbase