Senior ML Engineer: Synthetic Data Pipelines

Cohere; United States; Remote; USD 180,000 - 260,000; Full time

Job summary

Cohere is seeking a Senior Machine Learning Engineer specializing in synthetic data to lead the end-to-end management of the synthetic data pipeline for our foundation AI models. You will maintain, optimize, and evaluate data quality, transforming diverse web and code data with generative methods to boost token efficiency and model performance.

You will work with researchers and engineers across teams to bridge raw data and high-quality AI models, delivering improvements in throughput and

Qualifications

• Strong software engineering skills with Python.
• Experience building data pipelines and scalable systems.
• Familiarity with data processing frameworks like Spark/Beam.

Responsibilities

• Design and build scalable inference pipelines on large GPU clusters.
• Conduct data ablations and experiment with data mixtures to improve model performance.
• Research and implement synthetic data curation methods to advance NLP models.
• Collaborate with researchers and engineers to ensure data pipelines meet model requirements.

Skills

Python

Data pipelines

GPU Compute

Distributed Systems

Tools

Apache Spark

Apache Beam

Job description

Cohere is seeking a Senior Machine Learning Engineer specializing in synthetic data to lead the end-to-end management of the synthetic data pipeline for our foundation AI models. You will maintain, optimize, and evaluate data quality, transforming diverse web and code data with generative methods to boost token efficiency and model performance.

You will work with researchers and engineers across teams to bridge raw data and high-quality AI models, delivering improvements in throughput and