The opportunity
Cohere is the leading security-first enterprise AI company. We build cutting-edge foundation AI models and end-to-end products that are designed to solve real-world business problems.
What you'll do
Design and write high-performant and scalable software for training models.
Consistently post-train the models to reach SOTA level performance.
Coordinate with other specialist teams (Agentic, Code…) to produce models: that have strong all encompassing performance.
Craft and implement techniques to improve the performance and results of our: training cycles both on the SFT and the RL regime.
Research, implement, and experiment with ideas on our supercompute and data infrastructure.
Learn from and work with the best researchers in the field.
What they're looking for
- Extremely strong software engineering skills.
- Proficiency in Python and related ML frameworks such as JAX, Pytorch and XLA/MLIR.
- Experience with distributed training infrastructures (Kubernetes, Slurm) and associated frameworks (Ray).
- Experience using large-scale distributed training strategies.
