Member of Technical Staff, Agentic Environments
Remote in CanadaCohere: Cohere is the leading security-first enterprise AI company.
Candidates need extremely strong software engineering skills, proficiency in Python and ML frameworks like JAX or PyTorch, and experience debugging large-scale distributed training. Experience with Kubernetes, Ray and ML or LLM academic research is also required.
Python, PyTorch, JAX, Kubernetes, Ray
Research Internship (Winter 2027)
Toronto, ON +1Cohere: Cohere is the leading security-first enterprise AI company.
Candidates need to be pursuing or completing a PhD in machine learning, NLP or a related field, with experience in distributed training, transformers and Python, and strong communication skills. Cohere will also consider exceptional non-PhD candidates.
PyTorch, JAX, TensorFlow, Python, C++
Member of Technical Staff, Integration/RL Team (Research Engineer)
Toronto, ON +1Cohere: Cohere is the leading security-first enterprise AI company.
Candidates need very strong software engineering skills, proficiency in Python and ML frameworks such as JAX, PyTorch or XLA/MLIR, and experience debugging large-scale distributed training. Test-driven development and clean code habits are valued. Experience with Kubernetes, Ray, post-training and ML or RL research are bonus qualifications.
Python, JAX, PyTorch, XLA/MLIR, Kubernetes, Ray
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
Senior Member of Technical Staff, Multimodal AI
Toronto, ON +1Cohere: Cohere is the leading security-first enterprise AI company.
Candidates need exceptional software engineering skills, strong Python, and experience with deep learning frameworks such as JAX, PyTorch and TensorFlow. Experience with distributed training, autoregressive models and evaluating large multimodal models is also expected.
Python, JAX, PyTorch, TensorFlow, CUDA, HuggingFace