Advance model post-training with Python and large-scale distributed training at Cohere, working remotely or from listed offices. Ship frontier models into production. As a Member of Technical Staff, you will bridge research and production while advancing model post-training. You will write production software, support research, and collaborate across specialist teams. The work spans SFT, RL, supercompute, and data infrastructure. Key Responsibilities:
Design scalable software for model training Post-train models toward state-of-the-art performance Coordinate with Agentic and Code specialist teams Improve SFT and RL training cycles Experiment on supercompute and data infrastructure Requirements:
Extremely strong software engineering skills Proficiency in Python, JAX, PyTorch, and XLA/MLIR Experience with Kubernetes, Slurm, and Ray Experience using large-scale distributed training strategies Hands-on experience training large models at scale Build production machine learning systems while optimizing post-training performance across Cohere’s research and engineering efforts.
#J-18808-Ljbffr
Machine Learning Software Engineer in Toronto employer: Cohere
Cohere is an exceptional employer that fosters a dynamic and collaborative work culture, making it an ideal place for an Executive Assistant to thrive. With a strong focus on employee growth, we offer generous benefits including a weekly lunch stipend, comprehensive health coverage, and a robust education stipend, all while working in a cutting-edge AI environment in Toronto. Join us to be part of a passionate team dedicated to innovation and meaningful impact in the world of AI.
