Cohere

Member of Technical Staff, Agent Code

London, Remote. Posted 14 months ago

The role

You will drive the development of code LLMs and agent systems through research and engineering. You will implement novel ideas, design training strategies, and lead experiments on compute infrastructure.

What you will do

  • Stay up-to-date with research in code LLMs and agents
  • Implement novel ideas into systems
  • Design and implement scalable strategies to train code models
  • Deploy agent frameworks for inference and sampling
  • Collaborate with the pretraining team
  • Create SFT trajectories and work on existing and new RL algorithms
  • Hillclimb on existing benchmarks
  • Design new benchmarks for enterprise users
  • Lead experiments on state-of-the-art compute infrastructure

Required qualifications

  • PhD in Computer Science, Machine Learning, or a related field
  • Publications in top-tier venues (e.g., NeurIPS, ICML, ICLR, ACL, EMNLP)
  • Deep expertise in code LLMs and agent systems
  • Active contribution to the development of code models
  • Hands-on experience with frontier LLMs and their applications in code generation or automation
  • Strong software engineering skills
  • Proficiency in Python and PyTorch, TensorFlow, or similar frameworks
  • Experience with distributed systems, cloud infrastructure, and scalable architectures
  • Proactive, self-motivated mindset

Benefits

  • Weekly lunch stipend of $75/£75 or equivalent
  • Full health and dental benefits
  • Separate budget for mental health
  • RRSP matching, 401K, Pension Scheme
  • 100% Parental Leave top-up for up to 6 months
  • Annual enrichment benefits for arts, culture, fitness, wellness, quality time, and workspace improvement
  • Education and learning stipend for conferences, courses, and coaching
  • 6 weeks of paid vacation
  • Budget for traveling to other offices if remote
  • Annual company offsite
  • Daily lunch program and snacks for those in office
  • Co-working benefit for those not near an office

Before you apply

  • Candidates should be in locations that align with ET to CET time zones
  • May use AI-enabled tools to screen and assess applicants

About the company

Cohere is a security-first enterprise AI company that builds foundation AI models and end-to-end products. It is a global technology company headquartered in Toronto.