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

LinkedIn · 1 day ago

Staff Software Engineer, AI Platform

LinkedIn is the world’s largest professional network, committed to creating economic opportunities for every member of the global workforce. The Staff Software Engineer role focuses on building and optimizing AI model training and serving infrastructure, collaborating with teams to advance scalable AI platforms.

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

Responsibilities

Owning the technical strategy for broad or complex requirements with insightful and forward-looking approaches that go beyond the direct team and solve large open-ended problems
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
Mentoring other engineers, defining our challenging technical culture, and helping to build a fast-growing team
Working closely with the open-source community to participate and influence cutting edge open-source projects (e.g., vLLMs, PyTorch, GNNs, DeepSpeed, Huggingface, etc.)
Functioning as the tech-lead for several concurrent key initiatives AI Infrastructure and defining the future of AI Platforms

Qualification

Deep Learning SystemsDistributed SystemsMachine Learning InfrastructurePythonJavaC++GPU ServingTensorFlowPyTorchStakeholder ManagementCommunication

Required

Bachelor's Degree in Computer Science or related technical discipline, or equivalent practical experience
4+ years of experience in the industry with leading/ building deep learning systems
4+ years of experience with Java, C++, Python, Go, Rust, C# and/or Functional languages such as Scala or other relevant coding languages
Hands-on experience developing distributed systems or other large-scale systems

Preferred

BS and 8+ years of relevant work experience, MS and 7+ years of relevant work experience, or PhD and 4+ years of relevant work experience
Previous experience working with geographically distributed co-workers
Outstanding interpersonal communication skills (including listening, speaking, and writing) and ability to work well in a diverse, team-focused environment with other SRE/SWE Engineers, Project Managers, etc
Experience building ML applications, LLM serving, GPU serving
Experience with search systems or similar large-scale distributed systems
Expertise in machine learning infrastructure, including technologies like MLFlow, Kubeflow and large scale distributed systems
Experience with distributed data processing engines like Flink, Beam, Spark etc., feature engineering
Co-author or maintainer of any open-source projects
Familiarity with containers and container orchestration systems
Expertise in deep learning frameworks and tensor libraries like PyTorch, Tensorflow, JAX/FLAX

Benefits

Annual performance bonus
Stock
Benefits and/or other applicable incentive compensation plans
Generous health and wellness programs and time away for employees of all levels.

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
Bain Capital VenturesGreylockSequoia 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