Senior Member of Technical Staff, Web Data

As a Senior Member of Technical Staff specializing in web data for pre-training, you will play a pivotal role in developing the large scale web data pipeline that underpins Cohere’s advanced language models. In this role, you will work extensively with large-scale web corpora, transforming raw, noisy internet data into high-quality training data for pretraining. You will own key components of the data pipeline, including extraction, parsing, deduplication, and filtering. You will also analyze the composition and quality of web data, study its impact on downstream model performance, and collaborate closely with the broader data and evaluation teams to iterate on the training corpus.

Responsibilities

Own key components of the data pipeline, including extraction, parsing, deduplication, and filtering; Maintain large-scale pipelines for processing web corpora; Work on filtering and quality-scoring systems to identify high-value web documents; Analyze web data composition across domains, languages and time periods; Develop and maintain highly-performant deduplication pipelines; Collaborate with cross-functional teams, including researchers and engineers, to ensure data pipelines meet the demands of cutting-edge language models

Requirements

Strong software engineering skills, with proficiency in Python and experience building data pipelines; Familiarity with data processing frameworks such as Apache Spark, Apache Beam, Pandas, or similar tools; Experience working with large-scale web datasets; 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

Nice to have

Paper at top-tier venues (such as NeurIPS, ICML, ICLR, AIStats, MLSys, JMLR, AAAI, Nature, COLING, ACL, EMNLP)

Benefits

Weekly lunch stipend of $75/£75 or equivalent in local currency; 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 including arts & culture, fitness/wellness, quality time, workspace improvement credit, and education & learning stipend; 6 weeks of paid vacation (30 working days); Budget for traveling to other offices if remote, plus annual company offsite; Co-working benefit for those not near an office; $500 home office stipend; For those in the office: daily lunch program, snacks, and community and social events