GrepJob
Cohere

Member of Technical Staff, Agent Code

Cohere
Apply
8 months ago
Toronto, Canada +4 moreSenior / Staff+
H1B Sponsor

Responsibilities

  • Stay up-to-date with the latest research in code LLMs and agents.
  • Design and implement scalable strategies to train code models.
  • Deploy agent frameworks for inference and sampling.
  • Collaborate with the pretraining team to create SFT trajectories.
  • Lead experiments on state-of-the-art compute infrastructure.

Requirements

  • PhD in Computer Science, Machine Learning, or a related field.
  • Deep expertise in code LLMs and agent systems.
  • Hands-on experience with frontier LLMs in code generation or automation.
  • Strong software engineering skills, particularly in Python and PyTorch or TensorFlow.
  • Experience with distributed systems and scalable architectures.
  • Proactive and self-motivated mindset.

Benefits

  • Open and inclusive culture and work environment.
  • Weekly lunch stipend, in-office lunches, and snacks.
  • Full health and dental benefits, including mental health budget.
  • 100% Parental Leave top-up for up to 6 months.
  • Personal enrichment benefits for arts, culture, fitness, and workspace improvement.
  • Remote-flexible work options and co-working stipend.
  • 6 weeks of vacation (30 working days).

Tech Stack

PythonPyTorchTensorFlow

Categories

AI & MLBackendData Science