Software Engineer, Internal Infrastructure (North America)

About the Role

Our mission is to scale intelligence to serve humanity. We’re training and deploying frontier models for developers and enterprises who are building AI systems to power magical experiences like content generation, semantic search, RAG, and agents. We believe that our work is instrumental to the widespread adoption of AI. We obsess over what we build. Each one of us is responsible for contributing to increasing the capabilities of our models and the value they drive for our customers. We like to work hard and move fast to do what’s best for our customers. Cohere is a team of researchers, engineers, designers, and more, who are passionate about their craft. Each person is one of the best in the world at what they do. We believe that a diverse range of perspectives is a requirement for building great products. Join us on our mission and shape the future! The internal infrastructure team is responsible for building world-class infrastructure and tools used to train, evaluate and serve Cohere's foundational models. By joining our team, you will work in close collaboration with AI researchers to support their AI workload needs on the cutting edge, with a strong focus on stability, scalability, and observability. You will be responsible for building and operating Kubernetes GPU superclusters across multiple clouds. Your work will directly accelerate the development of industry-leading AI models that power Cohere's platform North.

Responsibilities

• Build and operate Kubernetes compute superclusters across multiple clouds
• Partner with cloud providers to optimize infrastructure costs, performance, and reliability for AI workloads
• Work closely with research teams to understand their infrastructure needs and identify ways to improve stability, performance, and efficiency of novel model training techniques
• Design and build resilient, scalable systems for training AI models, focusing on creating intuitive user interfaces that empower researchers to self-serve to troubleshoot and resolve problems
• Encourage software best practices across our company and participate in team processes such as knowledge sharing, reviews, and on-call

Requirements

• Deep experience running Kubernetes clusters at scale and/or scaling and troubleshooting Cloud Native infrastructure, including Infrastructure as Code
• Strong programming skills in Go or Python
• Prefer contributing to Open Source solutions rather than building solutions from the ground up
• Are self-directed and adaptable, and excel at identifying and solving key problems
• Draw motivation from building systems that help others be more productive
• See mentorship, knowledge transfer, and review as essential prerequisites for a healthy team
• Have excellent communication skills and thrive in fast-paced environments

Qualifications

• You've previously worked with ML training infrastructure and GPU workloads and have familiarity with RDMA networking
• You have expertise to support and troubleshoot low level Linux systems
• You have experience collaborating with research teams or machine learning engineers