Member of Technical Staff, Training Infra Engineer

Cohere is at the forefront of AI development, focusing on training and deploying frontier models for developers and enterprises. This role as a Member of Technical Staff, Training Infra Engineer offers a unique opportunity to work on cutting-edge AI infrastructure. You'll be responsible for designing and implementing high-performance training pipelines, optimizing infrastructure, and bridging the gap between research and production. The company boasts one of the highest compute-to-engineer ratios globally and maintains a flexible approach to engineering and research work.

The position offers the chance to work with state-of-the-art technology and some of the best researchers in the field. You'll be contributing to critical infrastructure that enables large-scale model training, working with technologies like Python, JAX, PyTorch, and Kubernetes. The role requires strong software engineering skills and experience with distributed training systems.

Cohere values diversity and maintains an inclusive work environment, with offices in major tech hubs like London, Toronto, San Francisco, and New York, while also supporting remote work. The company offers comprehensive benefits including health and dental coverage, mental health support, parental leave, and generous vacation time. This is an excellent opportunity for someone passionate about AI infrastructure who wants to make a significant impact in the field of machine learning and artificial intelligence.

The company's mission to "scale intelligence to serve humanity" drives their work in developing AI systems for content generation, semantic search, RAG, and agents. They emphasize customer value and maintain a fast-paced, innovation-focused environment where each team member contributes to advancing model capabilities.