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

LinkedIn ยท 16 hours ago

Principal Staff Software Engineer, AI Training Platform

LinkedIn is the worlds largest professional network, built to create economic opportunity for every member of the global workforce. The Principal Staff Software Engineer will lead and build the next-gen training infrastructure to power AI use cases, focusing on developing high performance AI Training pipelines and optimizing deep learning training.

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 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., PyTorch, GNNs, DeepSpeed, Huggingface, etc.)
Functioning as the tech-lead for several concurrent key initiatives for the Training Infrastructure and defining the future of AI training platforms

Qualification

Deep Learning SystemsDistributed SystemsObject-Oriented ProgrammingMachine Learning InfrastructureOpen Source ContributionsStakeholder ManagementML Algorithm DevelopmentBig Data

Required

BS/BA in Computer Science or related technical field or equivalent technical experience
7+ years of industry experience in software design, development, and algorithm related solutions
7+ years of experience programming in object-oriented languages such as Python, C++, Java, Go, Rust, Scala
5+ years of experience as an architect, or technical leadership position
5+ years of experience in the industry with leading / building deep learning systems
Hands-on experience developing distributed systems or other large-scale systems

Preferred

MS or PhD in Computer Science or related technical discipline
12+ years of experience in software design, development, and algorithm related solutions with at least 5 years of experience in a technical leadership position
12+ years of experience in an object-oriented programming language such as Python, C++, Java, Go, Rust, Scala
5+ years of experience with large-scale distributed systems and client-server architectures
Co-author or maintainer of any open-source projects
Expertise in machine learning infrastructure, including technologies like MLFlow, Kubeflow and large scale distributed systems
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

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