Software Engineer, ML Workload Platform

Cohere is seeking a full-time Software Engineer for its ML Workload Platform. The official posting lists this opening as remote in Canada. Cohere also lists San Francisco as a secondary location, so candidates outside Canada should check the employer’s own location choices and eligibility before applying. The team builds the internal infrastructure used to train, evaluate and serve foundation models, working closely with AI researchers and machine learning engineers.

What you would do

• Build and operate the Kubernetes platform that schedules training, evaluation and reinforcement-learning workloads, including capacity allocation, job failures and useful feedback to researchers.
• Develop networking and runtime isolation for workloads, including agent and sandbox environments.
• Work with data infrastructure engineers on storage for model checkpoints and other training artifacts.
• Investigate production failures and performance bottlenecks, and improve the platform’s reliability and ease of operation.
• Build automation, infrastructure as code and observability for systems that support model development.
• Participate in a compensated around-the-clock on-call rotation.

Experience Cohere is looking for

• Production experience with a job platform, batch system or Kubernetes controllers, especially scheduling, queues and fair use of shared capacity.
• Sound distributed-systems judgment about retries, partial failures and the limits of automation.
• Experience relevant to ML training workflows, reinforcement-learning sandboxes, container networking or large-scale checkpoint storage. Direct GPU-cluster experience is helpful but is not stated as mandatory.
• Software development in Python or Go and the ability to extend established open-source systems where appropriate.
• Ability to collaborate with researchers or ML engineers and drive improvements across teams. Cohere says it considers more than one experience level.

Location and benefits

The employer feed marks this role remote and gives Canada as its primary location; this WorkinVirtual listing is limited to applicants eligible for the Canadian opening. Cohere describes itself as remote-friendly and offers an office-travel budget, a coworking benefit for people away from an office and a home-office stipend. The official posting also describes health and dental benefits, retirement contributions, parental-leave support, learning benefits and paid vacation; availability can vary by location. It does not state a salary for this opening, so no pay amount is presented here. Review the current official requisition and apply through Cohere’s direct application page linked on this listing.