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
Machine Learning Intern/Co-op (Winter 2027)
Remote in CanadaCohere: Cohere is the leading security-first enterprise AI company.
Cohere needs candidates currently enrolled in a post-secondary program and available for a full-time 3-6 month term. Required skills include Python, ML frameworks such as TensorFlow and JAX, distributed training, and Transformer-based models.
Python, TensorFlow, TF-Serving, JAX, XLA/MLIR, CUDA
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
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