Member of Technical Staff, ML Workload Platform

🔥 3 minutes ago

🌐 Canada, United States – Remote

🇨🇦 Canada – Remote 🇺🇸 United States – Remote

💵 CA$300k - CA$420k / year

⏰ Full Time

🔴 Lead

🖥 Software Engineer

👻 Ghost score 22%

What it means: the chance that no one is really hiring for this job.How to read it: 60% or higher means we think this is a ghost. Lower scores are less of a worry.How we score it: age, reposts, open roles vs team size, how fast jobs close, and whether pay is listed.Fine print: an estimate from public data, not a judgment of the company. Real jobs can score high, so treat it as a hint, not a fact.

Apply Now

Find Similar Remote Jobs

Cohere

WebsiteLinkedInAll Job Openings

11 - 50 employees

🤖 Artificial Intelligence

🏢 Enterprise

☁️ SaaS

Artificial Intelligence • Enterprise • SaaS

Cohere is a leading enterprise AI platform optimized for generative AI, search and discovery, and advanced retrieval. The company offers AI-powered applications designed to augment and elevate the global workforce, helping businesses thrive in the AI era. Cohere provides solutions such as embedding and reranking models, allowing enterprises to efficiently retrieve information and build powerful applications. The company offers flexible deployment options for enterprise-grade AI, on any cloud or on-premises, and provides extensive developer resources and support. Cohere is committed to scaling intelligence to serve humanity, making intelligence abundant, affordable, and accessible.

📋 Description

• Own the platform that schedules and runs training, evaluation, and reinforcement-learning jobs on Kubernetes • Manage capacity allocation, job failures, and platform-cause visibility for researchers • Build networking and runtime isolation, including agent and sandbox environments • Collaborate with the data infrastructure team on checkpoint and training-artifact storage • Troubleshoot production failures and performance problems • Build automation, infrastructure-as-code, and observability for maintainability • Work with AI researchers and ML engineers on stability, scalability, and observability • Participate in a compensated 24x7 on-call rotation

🎯 Requirements

• Experience running a job platform, batch system, or Kubernetes controllers in production • Experience with scheduling, queues, and multi-tenant fairness • Distributed systems judgment, including retries, partial failure, and knowing what not to automate • Experience with ML training workflows, RL sandboxes, container networking (CNI), or checkpoint storage at scale • Direct GPU-cluster experience is a plus • Programming experience in Python or Go • Experience working with researchers, ML engineers, or teams with differing goals • Experience finding bottlenecks, proposing fixes, and driving implementation in a fast-moving environment • Availability for a compensated 24x7 on-call rotation • Authorization to work in the country of residence • Required to answer questions about Kubernetes at scale and ML/HPC infrastructure, including GPU/TPU clusters, JAX, PyTorch, TensorFlow, and high-performance computing environments

🏖️ Benefits

• Weekly lunch stipend of $75/£75 or equivalent in local currency • Full health and dental benefits • Separate mental health budget • RRSP matching • 401K • Pension Scheme • 100% parental leave top-up for up to 6 months, for either parent • Annual enrichment benefits for arts and culture, fitness/wellness, quality time, and workspace improvement • Education and learning stipend for conferences, courses, and coaching • 6 weeks of paid vacation (30 working days) • Budget for traveling to other offices if remote • Annual company offsite • Daily lunch program, snacks, and regular community and social events for office-based employees • Co-working benefit for employees not near an office • $500 home office stipend • Compensated 24x7 on-call rotation

📊 Check your resume score for this job

Improve your chances of getting an interview by checking your resume score before you apply.

Upload Resume