Member of Technical Staff, Agent Code
Toronto, ONCohere: Cohere is the leading security-first enterprise AI company.
Cohere needs candidates with a PhD in computer science, machine learning or a related field, with top-venue publications, deep code LLM and agent expertise, and strong Python and PyTorch or TensorFlow skills. Experience with distributed systems and cloud infrastructure is also required.
PyTorch, TensorFlow, Python
Member of Technical Staff, Training Performance Engineer
Toronto, ON +1Cohere: Cohere is the leading security-first enterprise AI company.
Candidates need extremely strong software engineering skills, proficiency in Python and ML frameworks such as JAX, PyTorch and XLA/MLIR, and experience writing GPU kernels with CUDA or Triton. Experience with large-scale distributed training and transformer sequence models is also required.
CUDA, Triton, PyTorch, JAX, XLA, MLIR