Software Engineer, ML Workload Platform

Cohere is the leading security-first enterprise AI company. We build cutting-edge foundation AI models and end-to-end products that are designed to solve real-world business problems.

We are looking for a Software Engineer to join our Internal Infrastructure team, which builds the infrastructure and tools used to train, evaluate, and serve Cohere's foundation models. You will work with AI researchers and ML engineers on stability, scalability, and observability.

Responsibilities:

• Own the platform that schedules and runs training, evaluation, and reinforcement-learning jobs on Kubernetes
• Build the networking and runtime isolation those workloads need, including agent and sandbox environments
• Work with the data infrastructure team on storage for checkpoints and other training artifacts
• Troubleshoot production failures and performance problems, and leave the system easier to operate than you found it
• Build automation, infrastructure-as-code, and observability so the platform stays maintainable

Requirements:

• Experience running a job platform, batch system, or Kubernetes controllers in production
• Distributed systems judgment: retries, partial failure, and knowing what not to automate
• Experience with ML training workflows, RL sandboxes, container networking (CNI), or checkpoint storage at scale
• Proficiency in Python or Go
• Experience working with researchers, ML engineers, or another team whose goals were not identical to yours

Benefits:

• A weekly lunch stipend of $75/£75 or equivalent in your local currency for lunch
• 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, for either parent
• Annual enrichment benefits: Arts & culture, fitness/wellness, quality time, and a workspace improvement credit
• 6 weeks of paid vacation (30 working days!)
• Budget for traveling to other offices if you are remote, plus an annual company offsite