Staff AI Platform Engineer — Lead Production Infra

Code Metal, Inc.

Washington (District of Columbia)

Hybrid

USD 205,000 - 250,000

Full time

6 days ago
Be an early applicant
Application generator

Get a reply from this employer — a resume and cover letter tailored to exactly what they’re hiring for.

Get past ATS filters

Benefits offered by this job

Health care plan with 100% premium
401k with 5% matching
Paid Time Off (uncapped vacation)
Flexible hybrid or remote work
Relocation assistance

Job summary

Code Metal, Inc. is seeking a Staff AI Platform Engineer to lead a four-person team and architect the AI enablement stack. You will guide production systems, model gateways, and agent harnesses, collaborating with Applied AI Research to validate platform decisions.

The role emphasizes engineering leadership, platform reliability, and building scalable, verifiable workflows for product teams. Hybrid/remote options are supported.

Qualifications

  • Production-grade Python and platform engineering fundamentals: API and service design, distributed systems, containers and Kubernetes, CI/CD, and testing.
  • Shipped production agentic systems with reliability considerations.
  • Experience with context engineering: retrieval-augmented generation, embeddings, vector or hybrid search, and memory for agents.

Responsibilities

  • Set the technical direction and architecture for Code Metal's AI platform and lead the team building it.
  • Deploy, benchmark, and tune production inference for open-weight models on vLLM, SGLang, and TensorRT-LLM.
  • Own the model gateway that teams use to reach self-hosted and commercial models with robust auth and routing.

Skills

Python
Kubernetes
APIs & services
Distributed systems
OpenTelemetry
PyTorch
Hugging Face
Staff leadership
Design docs
Experiment design

Tools

vLLM
SGLang
TensorRT-LLM

Job description

Code Metal, Inc. is seeking a Staff AI Platform Engineer to lead a four-person team and architect the AI enablement stack. You will guide production systems, model gateways, and agent harnesses, collaborating with Applied AI Research to validate platform decisions.

The role emphasizes engineering leadership, platform reliability, and building scalable, verifiable workflows for product teams. Hybrid/remote options are supported.

Get your free, confidential resume review.

or drag and drop your file here.

Similar jobs

Similar jobs worth comparing

Remote Staff AI Platform Engineer - Lead AI Infra
Remote Staff AI Platform Engineer - Lead AI Infra

Code Metal, Inc. • Boston (MA), Northern (KY)

Hybrid
USD 190,000 - 270,000
Health coverage
401k with match
Unlimited PTO
+2
Staff AI Platform Architect — Remote
Staff AI Platform Architect — Remote

Code Metal, Inc. • United States

Hybrid
USD 180,000 - 280,000
Health coverage
401(k) matching
Uncapped vacation
+2
Staff AI Platform Engineer - Lead AI Infra & Pipelines
Staff AI Platform Engineer - Lead AI Infra & Pipelines

Code Metal, Inc. • San Francisco (CA)

Hybrid
USD 205,000 - 250,000
Health care plan for employees
401k with matching
Paid Time Off
Lead AI Platform Engineer - Production ML Infra (Remote)
Lead AI Platform Engineer - Production ML Infra (Remote)

Code Metal, Inc. • Washington

Hybrid
USD 150,000 - 230,000
Health Insurance
Dental Insurance
Vision Insurance
+6
Staff AI Platform Engineer — Scale Production ML Infra
Staff AI Platform Engineer — Scale Production ML Infra

DAT Freight & Analytics • Seattle (WA)

Hybrid
USD 198,000 - 246,000
Medical, Dental, Vision
Paid time off
401k match
+3
Senior AI Platform Engineer - Production Systems
Senior AI Platform Engineer - Production Systems

People in AI • San Francisco (CA)

Hybrid
USD 140,000 - 200,000
Staff AI Platform Engineer (Fully Remote)
Staff AI Platform Engineer (Fully Remote)

Code Metal, Inc. • United States

Hybrid
USD 180,000 - 280,000
Health coverage
401(k) matching
Uncapped vacation
+2
Staff AI Platform Engineer: Architect Production Agent Systems
Staff AI Platform Engineer: Architect Production Agent Systems

Sapiom • San Francisco (CA)

On-site
USD 180,000 - 260,000
Staff AI Architect — Lead Production Agentic AI
Staff AI Architect — Lead Production Agentic AI

Modernizing Medicine, Inc. • Boca Raton (FL)

Hybrid
USD 170,000 - 250,000
Medical, dental, and vision benefits
401(k) with company matching
Paid time off & parental leave
+1
Staff AI Platform Engineer: Scale ML Infra
Staff AI Platform Engineer: Scale ML Infra

DAT Freight Solutions • Seattle (WA)

Hybrid
USD 198,000 - 246,000
Medical Insurance
Dental Insurance
Vision Insurance
+6