Senior AI Cluster Engineer – GPU Performance & Pre-Sales

Iframe; San Francisco (CA); On-site; USD 220,000 - 320,000; Full time

Benefits offered by this job

Equity

Competitive base salary band

Job summary

iframe.ai is seeking a Senior Customer Engineer in the US to own three to five reserved-capacity accounts, guiding training and inference performance end-to-end. You will collaborate with a sales engineer and an account manager to manage technical relationships from kickoff through renewal.

Five-plus years in distributed- or ML-systems engineering with 64+ GPU jobs, strong PyTorch and CUDA know-how, and the ability to discuss technical scope with executives are required.

Qualifications

• Five-plus years of distributed-systems or ML-systems engineering — production experience with 64+ GPU jobs is required.
• Strong PyTorch / FSDP / Megatron-LM debugging skills.
• Working knowledge of CUDA and at least one of Triton or CUTLASS.
• Strong communicator. You'll write four post-mortems a quarter and present at customer all-hands.
• Comfort with executive-level conversations on technical scope, timelines, and trade-offs.

Responsibilities

• Own three to five reserved-capacity accounts as their named customer engineer. Most are AI labs or AI-native scale-ups running 256–2048-GPU jobs.
• Profile distributed training jobs: NCCL collectives, gradient overlap, checkpoint cost, MFU. Make and defend specific recommendations.
• Tune kernels and configurations alongside the runtime team — your changes go through the same review process and ship in the same release train.
• Run the technical pre-sales for your accounts' expansions and renewals: pilot design, scope, success criteria, post-mortem.
• Lead the post-mortem on any P1 affecting your accounts; escalate root-cause work into the runtime, cluster, or platform teams.
• Carry the customer-engineering on-call rotation alongside runtime and cluster SRE — about one week per six.

Skills

Distributed systems

ML systems engineering

PyTorch

FSDP

Megatron-LM

Tools

CUDA

Triton

CUTLASS

NCCL

wandb

Job description

iframe.ai is seeking a Senior Customer Engineer in the US to own three to five reserved-capacity accounts, guiding training and inference performance end-to-end. You will collaborate with a sales engineer and an account manager to manage technical relationships from kickoff through renewal.

Five-plus years in distributed- or ML-systems engineering with 64+ GPU jobs, strong PyTorch and CUDA know-how, and the ability to discuss technical scope with executives are required.