Senior ML Platform Architect

The Briefed; Boston (MA); On-site; USD 150,000 - 190,000; Full time

Benefits offered by this job

Annual bonus

Equity

Health benefits

Job summary

Klaviyo is seeking a Senior Machine Learning Platform Engineer to build core platforms, tools, and infrastructure used by ML/AI engineers. You will drive training and deployment pipelines, unify feature data, and scale systems to support 167K+ customers and billions of events.

In this backend-heavy role, you will review designs, write high-impact code, and mentor teams while shaping the ML Platform roadmap and ensuring reliable production performance.

Qualifications

• 7+ years of experience building backend or distributed systems for AI/ML
• Proven track record in production-scale AI/ML systems
• Strong programming skills in Python (or similar)
• Experience with cloud infra and platform basics (AWS, Kubernetes)
• Familiarity with streaming/batch data processing at scale
• Ability to collaborate across ML, product, infra, security teams
• Excellent communication and leadership capabilities

Responsibilities

• Enable ML/AI teams to ship models and features faster and more safely
• Evolve ML platform architecture for reliable production use
• Improve reliability, observability, and cost efficiency of ML workloads
• Architect and scale feature storage and real-time retrieval for training and inference
• Lead technical execution, code/design reviews, and mentor engineers across teams
• Drive architectural direction for core ML platform components
• Partner with product and engineering leaders to define ML Platform roadmap

Skills

Python

AWS

Kubernetes

Spark

Ray Data

ML Platforms

System Design

Tools

CI/CD pipelines

Job description

Klaviyo is seeking a Senior Machine Learning Platform Engineer to build core platforms, tools, and infrastructure used by ML/AI engineers. You will drive training and deployment pipelines, unify feature data, and scale systems to support 167K+ customers and billions of events.

In this backend-heavy role, you will review designs, write high-impact code, and mentor teams while shaping the ML Platform roadmap and ensuring reliable production performance.