Member of Technical Staff - Sovereign AI
Toronto, ONCohere: Cohere is the leading security-first enterprise AI company.
Candidates need Canadian citizenship and eligibility for security clearance, plus extremely strong software engineering skills in Python and ML frameworks. Experience training, evaluating and using large language models, and with large-scale distributed GPU training, is required; publications at top venues are a bonus.
Python
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, 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