companyCohere logo

Member of Technical Staff, Training Performance Engineer

CohereLondon
On-site FullTime

Clicking Apply Now takes you to AutoApply where you can tailor your resume and apply.


Unlock Your Potential

Generate Job-Optimized Resume

One Click And Our AI Optimizes Your Resume to Match The Job Description.

Is Your Resume Optimized For This Role?

Find Out If You're Highlighting The Right Skills And Fix What's Missing

Qualifications

Proficiency in software engineering with a strong focus on performance optimization, experience with machine learning frameworks, expertise in low-level programming (CUDA, Triton), and a solid understanding of data infrastructure and supercomputing environments. A collaborative spirit and eagerness to learn from industry leaders are essential.

About the job

Who are we?

At Cohere, our mission is to harness and scale intelligence for the betterment of humanity. We are dedicated to training and deploying cutting-edge models that empower developers and enterprises to create astonishing AI-driven experiences such as content generation, semantic search, retrieval-augmented generation (RAG), and intelligent agents. We believe that our innovations play a crucial role in the mainstream adoption of AI technologies.

We take great pride in the products we develop. Every team member contributes to enhancing our models' capabilities and the value they deliver to our customers. Our culture is driven by hard work and a commitment to swift execution that prioritizes our clients' needs.

Cohere is composed of a diverse team of researchers, engineers, designers, and more, all of whom are passionate about their respective fields. Each individual is among the best globally in their area of expertise. We firmly believe that a wide array of perspectives is essential for creating outstanding products.

Join us in our quest to shape the future of AI!

Why this role?

As a Performance Engineer within our Pre-Training team, you will play a pivotal role in optimizing the efficiency and performance of our advanced language models and systems. Your primary focus will be on enhancing critical model training metrics, including training throughput and maximizing accelerator utilization.

This team merges software engineering, machine learning, and low-level kernel design expertise to develop robust systems that elevate model performance. You will actively identify and eliminate performance bottlenecks, while also creating innovative training and profiling tools that support Cohere's mission of delivering efficient and reliable language understanding and generation capabilities, thereby driving advancements in natural language processing.

Note: While we have offices in London, Toronto, New York, and San Francisco, we are also remote-friendly! This team primarily operates between Eastern Time (ET) and Central European Time (CET), so we are looking for candidates in locations aligned with these time zones for effective collaboration.

As a Technical Staff Member, your responsibilities will include:

  • Crafting and developing high-performance, scalable software solutions for model training.

  • Comprehending architectural modifications and design choices and their impacts on training throughput and quality.

  • Writing low-level CUDA and Triton kernels to maximize performance from our accelerators.

  • Conducting research, implementation, and experimentation on our supercomputing and data infrastructure.

  • Collaborating with and learning from top researchers in the field.

About Cohere

Cohere is at the forefront of AI innovation, dedicated to scaling intelligence for the benefit of all. Our team, comprised of exceptional researchers, engineers, and designers, works tirelessly to develop pioneering models that enhance human-computer interactions. We foster an inclusive environment that thrives on diverse perspectives, driving our commitment to create transformative AI solutions.

Similar jobs

Tailoring 0 resumes

We'll move completed jobs to Ready to Apply automatically.