AI Infra Engineer: Scale ML Clusters Kubernetes & Slurm

Perplexity AI Inc.; San Francisco (CA); On-site; USD 190,000 - 250,000; Full time

Benefits offered by this job

Comprehensive health insurance

401(k) plan

Equity options

Job summary

A leading AI solutions provider in San Francisco is seeking an AI Infra Engineer to design and manage scalable Kubernetes clusters and optimize AI training infrastructure. The ideal candidate will have strong Kubernetes and Slurm administration skills, proficiency in Python and C++, and experience with ML frameworks like PyTorch. This position offers a competitive salary range from $190,000 to $250,000, along with comprehensive benefits, including health insurance and a 401(k) plan.

Qualifications

• Strong expertise in Kubernetes administration.
• Hands‑on experience with Slurm workload management.
• Experience deploying distributed training systems.
• Deep understanding of container orchestration.
• High familiarity with LLM architecture.

Responsibilities

• Design and maintain scalable Kubernetes clusters.
• Manage Slurm-based HPC environments.
• Develop APIs and orchestration systems.
• Implement resource scheduling and job management.
• Benchmark system performance and optimize infrastructure.

Skills

Kubernetes administration

Slurm workload management

Python programming

C++ programming

ML frameworks (PyTorch)

Container orchestration

API development

Monitoring and debugging

Tools

Kubernetes

Slurm

AWS

Job description

A leading AI solutions provider in San Francisco is seeking an AI Infra Engineer to design and manage scalable Kubernetes clusters and optimize AI training infrastructure. The ideal candidate will have strong Kubernetes and Slurm administration skills, proficiency in Python and C++, and experience with ML frameworks like PyTorch. This position offers a competitive salary range from $190,000 to $250,000, along with comprehensive benefits, including health insurance and a 401(k) plan.