Member of Technical Staff – Agent Code

About the role

Cohere is a security‑first enterprise AI company building cutting‑edge foundation models and end‑to‑end products. The Member of Technical Staff – Agent Code will work at the intersection of research and engineering to develop advanced code‑generating LLMs and autonomous agent systems for enterprise customers.

Key responsibilities

• Stay up‑to‑date with the latest research in code LLMs, agents, and related fields, implementing novel ideas into our systems.
• Design and implement scalable strategies to train code models, and deploy agent frameworks for inference and sampling.
• Collaborate with the pretraining team, create SFT trajectories and work on existing and new RL algorithms.
• Hill‑climb on existing benchmarks and design new ones that reflect the needs of our enterprise users.
• Lead experiments on our state‑of‑the‑art compute infrastructure, pushing the boundaries of what’s possible with frontier LLMs.

Required profile

• A PhD in Computer Science, Machine Learning, or a related field, with publications in top‑tier venues (e.g., NeurIPS, ICML, ICLR, ACL, EMNLP).
• Deep expertise in code LLMs and agent systems, with a strong understanding of the latest research and trends.
• Hands‑on experience with frontier LLMs and their applications in code generation or automation.
• Strong software engineering skills, with proficiency in Python and PyTorch, TensorFlow, or similar frameworks.
• Experience with distributed systems, cloud infrastructure, and scalable architectures.
• Proactive, self‑motivated mindset with a passion for solving ambitious, open‑ended problems.

Required skills

• Python
• PyTorch
• TensorFlow
• Distributed systems
• Cloud infrastructure

What we offer

• Weekly lunch stipend of $75/£75 (or equivalent).
• Full health and dental benefits, including a mental‑health budget.
• RRSP matching, 401(k), pension scheme.
• 100 % parental‑leave top‑up for up to 6 months.
• Six weeks of paid vacation (30 working days).
• Education and learning stipend for conferences, courses, and coaching.
• Home office stipend of $500 and co‑working benefit for remote workers.
• Travel budget to visit other offices and annual company off‑site.