Staff/Senior Staff Software Engineer, Agent Infrastructure

The role

You will build the next generation of agentic AI infrastructure. This includes creating platform capabilities for execution, storage, and state management to power autonomous AI agents at scale.

What you will do

• Turn emerging ML research ideas into production-ready infrastructure
• Build core platform capabilities for execution, storage, and state management
• Prototype and evaluate new technologies
• Partner with research teams to shape infrastructure

Required qualifications

• Experience building production ML infrastructure with strong systems fundamentals
• Hands-on work with agentic systems, multi-agent workflows, or agent development frameworks
• Familiarity with model routing and LLM provider frameworks across different model types and environments
• Experience with scalable, fault-tolerant distributed systems and Kubernetes
• A track record of moving quickly on prototypes and making good decisions about productionization

Nice to have

• Experience across on-prem, private cloud, and public cloud environments
• Familiarity with storage systems, embedded databases, or filesystem abstractions
• Experience with code execution sandboxes such as gVisor, Firecracker, Kata, or WASM runtimes
• Interest in emerging ML infrastructure, edge inference, or browser-native models
• Open-source contributions to LLM or agent infrastructure projects
• Experience with identity, workload auth, or capability-based security systems

Benefits

• Weekly lunch stipend of $75/£75 or equivalent
• Full health and dental benefits, including a separate budget for mental health
• RRSP matching, 401K, Pension Scheme
• 100% Parental Leave top-up for up to 6 months
• Annual enrichment benefits for arts & culture, fitness/wellness, quality time, and workspace improvement
• Education & learning stipend for conferences, courses, and coaching
• 6 weeks of paid vacation
• Budget for traveling to other offices if remote
• Annual company offsite
• Daily lunch program and snacks for those in office
• Co-working benefit for those not near an office
• $500 home office stipend

About the company

Cohere is a leading security-first enterprise AI company that builds foundation AI models and end-to-end products. It is a global technology company headquartered in Toronto.