Member of Technical Staff, Pre-Training Data

Job summary

Responsibilities

As a Machine Learning Engineer specializing in pretraining data, you will play a pivotal role in developing the data pipeline that underpins Cohere’s advanced language models. In this role, you will conduct data ablations to evaluate data quality and construct pre-training data mixtures to enhance…

• Conduct data ablations to assess data quality and experiment with data mixtures to enhance model performance.
• Develop robust data modeling techniques to ensure datasets are structured and formatted for optimal training efficiency.
• Research and implement innovative data curation methods, leveraging Cohere’s infrastructure to drive advancements in natural language processing.
• Collaborate with cross-functional teams, including researchers and engineers, to ensure data pipelines meet the demands of cutting-edge language models.

What they're looking for

• Strong software engineering skills, with proficiency in Python and experience building data pipelines.
• Familiarity with curriculum learning, data mixing and data attribution.
• Familiarity with data processing frameworks such as Apache Spark, Apache Beam, Pandas, or similar tools.
• Experience working with large-scale datasets, including web data, code data, and multilingual corpora.
• Knowledge of data quality assessment techniques and experimentation with data mixtures.

Tech stack

• Python
• Spark
• Pandas