Member of Technical Staff, ML Workload Platform (AI)

Location: Remote in Canada or the United States

Salary: Canada CA$300K–CA$420K plus equity; United States $235K–$335K plus equity.

Company

Cohere is an enterprise AI company building foundation models and end-to-end products for business applications.

What you will do

• Own the Kubernetes platform that schedules and runs training, evaluation, and reinforcement-learning jobs, including capacity allocation, failure handling, and platform diagnostics.
• Build networking and runtime isolation for workloads, including agent and sandbox environments.
• Collaborate with data infrastructure engineers on storage for checkpoints and training artifacts.
• Troubleshoot production failures and performance issues while improving operability.
• Build automation, infrastructure-as-code, and observability for a maintainable platform.
• Participate in a compensated 24x7 on-call rotation.

Requirements

• Production experience running a job platform, batch system, or Kubernetes controllers, including scheduling, queues, and multi-tenant fairness.
• Demonstrated distributed-systems judgment involving retries, partial failure, and automation trade-offs.
• Experience with ML training workflows, RL sandboxes, container networking, or checkpoint storage at scale.
• Software development experience in Python or Go and a preference for extending established open-source projects.
• Experience collaborating with researchers, ML engineers, or teams with differing goals.
• Experience identifying bottlenecks, proposing fixes, and driving implementation in a fast-moving environment.

Nice to have

• Direct GPU-cluster experience.

Culture & Benefits

• Remote-friendly work with access to offices and co-working benefits.
• Compensated 24x7 on-call rotation.
• Health and dental benefits, mental-health budget, retirement or pension matching, and up to six months of fully topped-up parental leave.
• Six weeks of paid vacation, annual enrichment benefits, and education and learning support.
• Home-office stipend, lunch stipend, office-travel budget for remote employees, and an annual company offsite.