AI Platform Engineer: Scale Large Models

LinkedIn; Mountain View (CA); Hybrid; USD 120,000 - 195,000; Full time

Job summary

LinkedIn is hiring for an AI Training Infra engineer to scale training, feature engineering, and serving for billions‑parameter models. You will design high‑performance data I/O, work with PyTorch and open source libraries, enable distributed training across hundreds of billions of parameters, and optimize GPU and container infrastructure.

The role emphasizes collaboration with researchers and engineers to advance one of the world’s most scalable AI platforms, while advancing your expertise in

Qualifications

• Bachelor’s degree in Computer Science or a related technical discipline or equivalent practical experience.
• 1+ years in the industry with leading/building deep learning systems.
• Experience with Java, C++, Python, Go, Rust, C# and/or Scala or other relevant coding languages.
• Experience, qualifications in Machine Learning, AI.

Responsibilities

• Design, implement, and optimize 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

Distributed systems

Java

Python

Go

Rust

C#

Scala

Education

Bachelor's degree in CS or related field

MS/PhD in CS or related technical discipline

Tools

PyTorch

TensorFlow

HuggingFace

Horovod

DeepSpeed

PyTorch Lightning

CUDA

cuDNN

NCCL

Kubernetes

Spark

Beam

Flink

Job description

LinkedIn is hiring for an AI Training Infra engineer to scale training, feature engineering, and serving for billions‑parameter models. You will design high‑performance data I/O, work with PyTorch and open source libraries, enable distributed training across hundreds of billions of parameters, and optimize GPU and container infrastructure.

The role emphasizes collaboration with researchers and engineers to advance one of the world’s most scalable AI platforms, while advancing your expertise in