← All roles

Member of Technical Staff, Pre-Training Data

Cohere· Modeling
Apply on company site
Remote$205k - $535kFull-timePosted 11mo ago

About the role

Machine Learning Engineer specializing in pretraining data responsible for building data pipelines, conducting data ablations, and developing data mixtures to improve language model training efficiency and performance.

Responsibilities

  • 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.

Qualifications

  • 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.
  • A passion for bridging research and engineering to solve complex data-related challenges in AI model training.

Skills

Required:PythonData pipelinesData processing frameworks
Nice to have:Curriculum learningData mixing

Benefits

  • A weekly lunch stipend of $75/£75 or equivalent in your local currency for lunch.
  • Full health and dental benefits, including a separate budget for mental health.
  • RRSP matching, 401K, Pension Scheme.
  • 100% Parental Leave top-up for up to 6 months, for either parent.
  • Annual enrichment benefits: Arts & culture, fitness/wellness, quality time, and a workspace improvement credit; Education & learning stipend for conferences, courses, and coaching.
  • 6 weeks of paid vacation (30 working days).
  • Budget for traveling to other offices if you are remote, plus an annual company offsite.
  • $500 home office stipend to set up your workspace properly.
Last updated Oct 21, 2026. Interested? Apply on the company site.
Cohere logo

About Cohere

Support automation that's actually intelligent

Cohere resolves 40-70% of support requests using AI, over all channels, within just a few days.

artificial-intelligenceautomationliquid-ai-recruiting-target
Company profileWebsite

More open roles at Cohere