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 - Post-Training (Reinforcement Learning & Generalization)
Toronto, ONCohere: Cohere is the leading security-first enterprise AI company.
Candidates need strong research and software engineering skills, hands-on experience training LLMs, and experience applying reinforcement learning to LLMs. A record of work on generalization, robustness or distribution shift, and rigorous experimental practice are also required.
JAX, PyTorch, FSDP, Megatron
Member of Technical Staff, Pre-Training Data
Toronto, ON +1Cohere: Cohere is the leading security-first enterprise AI company.
Candidates need strong Python and data pipeline engineering skills, familiarity with data mixing and data processing frameworks, and experience with large-scale web, code and multilingual datasets. A research-oriented mindset bridging research and engineering is valued.
Python, Apache Spark, Apache Beam, Pandas
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