Member of Technical Staff, Agent Code

Role overview

Cohere is a security-first enterprise AI company building foundation models and products that solve real-world business challenges. This position brings together research and engineering to advance code-generating large language models and autonomous agents. You will lead development of cutting-edge systems that enable software automation and task execution with minimal human intervention, working alongside talented researchers and engineers with access to world-class compute resources.

Key responsibilities

Monitor developments in code LLMs, agents, and adjacent research areas, translating novel findings into production systems.
Design and implement scalable training strategies for code models and deploy agent frameworks for inference and sampling.
Collaborate with the pretraining team to create supervised fine-tuning trajectories and develop new reinforcement learning algorithms.
Evaluate performance on existing benchmarks and create new benchmarks aligned with enterprise user needs.
Conduct experiments using state-of-the-art compute infrastructure to expand the capabilities of frontier large language models.

Requirements

PhD in Computer Science, Machine Learning, or a related discipline with publications in top-tier venues such as NeurIPS, ICML, ICLR, ACL, or EMNLP.
Deep expertise in code LLMs and agent systems, demonstrated through active contribution to their development rather than passive familiarity.
Hands-on experience deploying frontier LLMs for code generation and automation applications.
Strong software engineering capabilities with proficiency in Python and PyTorch, TensorFlow, or equivalent frameworks.
Experience working with distributed systems, cloud infrastructure, and scalable system design.
Self-motivated approach to tackling complex, open-ended research challenges.

What makes this role worth considering

This role operates primarily across ET to CET time zones with flexible remote and office options, including locations in London, Toronto, New York, and San Francisco. Cohere offers comprehensive benefits including a weekly lunch stipend, full health and dental coverage with dedicated mental health support, six weeks of paid vacation, 100% parental leave top-up for six months, professional development budgets, and an annual company offsite with travel support for remote employees. You’ll work on problems at the frontier of AI with infrastructure and resources designed to enable exceptional work.