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 trainingPost-train models toward state-of-the-art performanceCoordinate with Agentic and Code specialist teamsImprove SFT and RL training cyclesExperiment on supercompute and data infrastructureRequirements: Extremely solid software engineering skillsProficiency in Python, JAX, Py Torch, and XLA/MLIRExperience with Kubernetes, Slurm, and RayExperience using large-scale distributed training strategiesHands-on experience training large models at scaleBuild production machine learning systems while optimizing post-training performance across Cohere’s research and engineering efforts.#J-18808-Ljbffr

📌 Machine Learning Software Engineer (Toronto)
🏢 Cohere
📍 Toronto