Member of Technical Staff, Agent Code

Stack

Responsibilities

• ▹Stay up to date with the latest research in code LLMs and agents, 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
• ▹Hillclimb on existing benchmarks and design new ones reflecting enterprise user needs
• ▹Lead experiments on state-of-the-art compute infrastructure, pushing the boundaries of frontier LLMs

Requirements

• ▹A PhD in Computer Science, Machine Learning, or a related field, with publications at top-tier venues (NeurIPS, ICML, ICLR, ACL, EMNLP)
• ▹Deep expertise in code LLMs and agent systems, with strong understanding of the latest research and trends
• ▹Active contribution to the development of code models, not just experience working with them
• ▹Hands-on experience with frontier LLMs 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
• ▹A proactive, self-motivated mindset with a passion for solving ambitious, open-ended problems

Soft skills

What we offer

• ▹Weekly lunch stipend of about $75/£75 (or local equivalent)
• ▹Full health and dental benefits, with a separate mental health budget
• ▹RRSP matching, 401K, or pension scheme
• ▹100% parental leave top-up for up to 6 months, for either parent
• ▹Annual enrichment benefits: arts & culture, fitness/wellness, quality time, workspace improvement
• ▹Education & learning stipend for conferences, courses, and coaching
• ▹6 weeks (30 working days) of paid vacation
• ▹Budget for traveling to other offices, plus an annual company offsite
• ▹$500 home office stipend; co-working benefit for remote employees

About the company

Cohere is the leading security-first enterprise AI company, building cutting-edge foundation models and end-to-end products that solve real business problems. Co-headquartered in Toronto and San Francisco, with key offices in London, New York City, Montreal, Seoul, Germany and Paris.