Senior Staff AI Infrastructure Engineer – Agent Platforms

Wwshemi; United States; Hybrid; USD 180,000 - 240,000; Full time

Benefits offered by this job

Lunch stipend

Health and dental benefits

RRSP matching / 401K

Parental Leave

Enrichment benefits

Learning stipend

Paid vacation

Home office stipend

Job summary

Cohere is building the next generation of agentic AI infrastructure. This role sits at the intersection of ML systems, distributed infrastructure, and developer experience, creating the platform that powers autonomous AI agents at scale.

You’ll work on hard problems with few established patterns, including secure code execution, agent state management, model routing, and resource management for long-running agent workflows.

Qualifications

• Experience building production ML infrastructure with strong systems fundamentals.

Responsibilities

• Turn emerging ML research ideas into production-ready infrastructure.
• Build core platform capabilities for execution, storage, and state management.
• Prototype and evaluate new technologies, then help decide what should move into production.
• Partner with research teams to shape infrastructure based on what future agent systems will need.

Skills

ML infrastructure

Systems fundamentals

Agentic systems

Kubernetes

Tools

Kubernetes

Job description

Cohere is building the next generation of agentic AI infrastructure. This role sits at the intersection of ML systems, distributed infrastructure, and developer experience, creating the platform that powers autonomous AI agents at scale.

You’ll work on hard problems with few established patterns, including secure code execution, agent state management, model routing, and resource management for long-running agent workflows.