Machine Learning Engineer • Senior

Who you are

You have 5+ years of experience in machine learning systems engineering, with a strong focus on building and maintaining frameworks for large-scale model training. Your expertise in Python and familiarity with machine learning libraries like TensorFlow and PyTorch enable you to contribute effectively to complex projects. You understand distributed systems and have experience working with high-performance computing (HPC) infrastructure, which allows you to design scalable solutions that meet the demands of modern AI applications.

You are comfortable working across the full stack of ML systems, from data processing to model deployment. Your ability to collaborate with researchers and engineers ensures that you can translate innovative ideas into practical tools that enhance model training efficiency. You thrive in environments that require quick problem-solving and adaptability, and you are committed to delivering high-quality results that drive value for customers.

Desirable

Experience with Kubernetes and Docker is a plus, as these tools are essential for managing containerized applications in a distributed environment. Familiarity with cloud platforms such as AWS or GCP will also be beneficial, as you will be working with cloud-based resources to scale training processes.

What you'll do

In this role, you will build and own the training framework responsible for large-scale language model training. You will design distributed training abstractions that optimize data flow and processing efficiency, ensuring that our models can be trained quickly and reliably. Your work will involve collaborating with cross-functional teams to connect research ideas to thousands of GPUs, enabling rapid experimentation and deployment of new features.

You will also be responsible for maintaining and evolving the core components of our training infrastructure, ensuring that they remain robust and scalable as our models grow in complexity. Your contributions will directly impact the capabilities of our models and the experiences they provide to users.

What we offer

Cohere provides a supportive and inclusive work environment where you can thrive. We offer a competitive salary and benefits package, including a generous vacation policy and parental leave top-up. Our commitment to mental health and personal enrichment ensures that you have the resources you need to maintain a healthy work-life balance. With offices in London, Toronto, New York, San Francisco, and Paris, we also offer flexible remote work options to accommodate your lifestyle.