Machine Learning Engineer • Mid-Level

Who you are

You have a strong background in machine learning and data engineering, with experience in developing data pipelines that support advanced AI models. Your expertise in Python allows you to efficiently manipulate and analyze large datasets, ensuring high data quality for model training. You understand the intricacies of natural language processing (NLP) and are familiar with frameworks like TensorFlow and PyTorch, which you have used to build and optimize machine learning models.

You are detail-oriented and analytical, capable of conducting data ablations to evaluate data quality and construct effective pre-training data mixtures. Your ability to bridge the gap between raw data and cutting-edge AI models is essential, as you contribute to improvements in critical training metrics like throughput and accelerator utilization. You thrive in collaborative environments, working alongside researchers and engineers to drive innovation in language understanding and generation capabilities.

Desirable

Experience with large-scale data processing and familiarity with cloud platforms such as AWS or GCP would be advantageous. A background in research or a strong understanding of the latest advancements in AI and machine learning will set you apart.

What you'll do

As a Machine Learning Engineer at Cohere, you will play a pivotal role in developing and optimizing the data pipeline that underpins our advanced language models. You will work closely with cross-functional teams to ensure that the data used for training is of the highest quality, directly impacting the performance of our models. Your responsibilities will include conducting data ablations to assess data quality and constructing pre-training data mixtures that enhance model performance.

You will collaborate with researchers to implement innovative solutions that improve training efficiency and effectiveness. Your work will involve analyzing training metrics and making data-driven decisions to refine our processes. You will also contribute to the development of tools and frameworks that facilitate the integration of new data sources into our training pipeline.

What we offer

Cohere provides a supportive and inclusive work environment where you can thrive. We offer a flexible remote work policy, allowing you to balance your professional and personal life effectively. Our benefits include a generous vacation policy, mental health support, and personal enrichment benefits that promote your well-being and professional growth. Join us in our mission to scale intelligence and shape the future of AI.