Data Engineer • Mid-Level

Who you are

You have a strong background in data engineering, with experience in managing end-to-end data pipelines — you've worked with diverse data sources and understand the importance of data quality and reliability. Your expertise in Python and SQL allows you to efficiently manipulate and analyze large datasets, ensuring they are structured for optimal model performance.

You are familiar with tools like Airflow and Apache Spark, which you have used to automate data workflows and process large volumes of data — your ability to bridge the gap between raw data and AI models is a key strength. You thrive in collaborative environments, working closely with researchers and engineers to enhance the capabilities of AI systems.

What you'll do

In this role, you will be responsible for developing and maintaining the data pipeline that supports Cohere's advanced language models — this includes tasks such as data ingestion, cleaning, filtering, and optimization. You will ensure that datasets are structured and formatted correctly, working with various data types including web data, code data, and multilingual corpora.

You will collaborate with cross-functional teams to identify data needs and implement solutions that enhance model performance — your contributions will directly impact the quality and diversity of the training data used in AI systems. You will also monitor data quality and reliability, making adjustments as necessary to improve the overall data pipeline.

What we offer

Cohere provides a supportive work environment that values mental health and well-being — we offer a comprehensive parental leave top-up and personal enrichment benefits. You will enjoy a flexible work schedule with the option to work remotely or from our offices in Toronto, New York, San Francisco, London, and Paris. We also provide generous vacation time, allowing you to recharge and maintain a healthy work-life balance.