Cohere is seeking an exceptional software engineer to advance model training and post-training techniques. You will design scalable software for training models, push SOTA results post-training, and collaborate with cross-functional teams to deliver production-ready architectures.
This role blends research and engineering across compute-heavy workflows, with remote-friendly options and global offices. You will work with Python, JAX, PyTorch, and MLIR tooling, and engage with Kubernetes, Slurm,
#J-18808-Ljbffr
Staff ML Engineer — Production-Ready AI Models employer: Cohere
Cohere is an exceptional employer that fosters a dynamic and inclusive work culture, where innovation thrives and employees are empowered to make a real impact in the AI landscape. With generous benefits such as a weekly lunch stipend, comprehensive health coverage, and a robust education stipend, we prioritise employee well-being and growth. Our London office offers a collaborative environment with opportunities for meaningful engagement with enterprise clients, making it an ideal place for those looking to advance their careers in cutting-edge technology.
