Member of Technical Staff, Data Analysis and Evaluation
Toronto, ON +1Cohere: Cohere is the leading security-first enterprise AI company.
Cohere needs candidates with extremely strong software engineering skills, statistical expertise in evaluating experiments, and experience designing data collection with human annotators. Candidates also need hands-on LLM training on distributed infrastructure, proficiency in Python and ML frameworks, and strong communication skills. One or more papers at top-tier venues are listed as a qualification.
Python, PyTorch, TensorFlow, JAX
Member of Technical Staff, Post-Training
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 like JAX and PyTorch, and hands-on experience training large models with distributed infrastructure. Post-training experience with a focus on performance optimisation is essential.
Kubernetes, Slurm, Ray, JAX, PyTorch, XLA
Member of Technical Staff, Training Infra Engineer
Toronto, ON +1Cohere: Cohere is the leading security-first enterprise AI company.
Candidates need strong software engineering skills, proficiency in Python and ML frameworks such as JAX, PyTorch and XLA/MLIR, and experience with distributed training on Kubernetes, Slurm and Ray. Hands-on experience training large models at scale and contributing to training infrastructure is required.
Python, JAX, PyTorch, XLA, MLIR, Kubernetes
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
Member of Technical Staff, Modeling
Toronto, ON +1Cohere: Cohere is the leading security-first enterprise AI company.
Candidates need extremely strong software engineering skills and proficiency in Python and ML frameworks. Experience writing GPU kernels with CUDA, large-scale distributed training and autoregressive sequence models such as Transformers is also expected.
Python, TensorFlow, TF-Serving, JAX, XLA, MLIR