Senior Member of Technical Staff, Synthetic Data

Stack

Responsibilities

• ▹Design and build scalable inference pipelines that run on large GPU clusters
• ▹Conduct data ablations to assess data quality and experiment with data mixtures to enhance model performance
• ▹Research and implement innovative synthetic data curation methods, leveraging Cohere's infrastructure
• ▹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 or Pandas
• ▹Experience working with LLMs through work projects, open-source contributions or personal experimentation
• ▹Familiarity with LLM inference frameworks such as vLLM and TensorRT
• ▹Experience working with large-scale datasets, including web data, code data and multilingual corpora
• ▹A passion for bridging research and engineering to solve complex data-related challenges in AI model training

Nice to have

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

Soft skills

What we offer

• ▹Weekly lunch stipend ($75/£75 or equivalent in local currency)
• ▹Full health and dental benefits, including a separate mental health budget
• ▹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, workspace improvement)
• ▹Education and learning stipend for conferences and courses
• ▹6 weeks of paid vacation (30 working days)
• ▹Travel budget to other offices for remote employees, plus an annual company offsite
• ▹$500 home office stipend

About the company

Cohere is a leading security-first enterprise AI company building cutting-edge foundation AI models and end-to-end products that solve real-world business problems for enterprises building AI systems. Offices in London, Paris, Toronto, San Francisco and New York, but remote-friendly: there are no location restrictions for this role between EST and EU.