[Hiring] Member of Technical Staff, ML Workload Platform @Cohere

Role Description

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.

You will work with AI researchers and ML engineers on stability, scalability, and observability. The work accelerates the models behind Cohere's platform, North.

• 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.

Qualifications

• Run a job platform, batch system, or Kubernetes controllers in production.
• Shown distributed systems judgment: retries, partial failure, and knowing what not to automate.
• Worked on ML training workflows, RL sandboxes, container networking (CNI), or checkpoint storage at scale.
• Written software in Python or Go.
• Worked with researchers, ML engineers, or another team whose goals were not identical to yours.
• Found a bottleneck, proposed a fix, and driven it in a fast-moving environment.

Requirements

• Direct GPU-cluster experience is a plus.

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.
• Education & learning stipend for conferences, courses, and coaching.
• 6 weeks of paid vacation (30 working days!).
• Budget for traveling to other offices if you are remote, plus an annual company offsite.

Company Description

Cohere is a global technology company headquartered in Toronto with key offices in London, New York City, San Francisco, Montreal, Paris, Berlin, and Seoul.

We strive to create an inclusive work environment for all; we welcome applicants from all backgrounds and are committed to providing equal opportunities.