Performance Engineer • Mid-Level

Who you are

You have a strong background in performance engineering, particularly in optimizing machine learning models and systems. Your experience includes working with advanced language models, and you understand the intricacies of training throughput and accelerator utilization. You are proficient in Python and have hands-on experience with machine learning frameworks such as TensorFlow. Your skills extend to low-level kernel design and development, allowing you to identify and remove performance bottlenecks effectively.

You thrive in collaborative environments, working closely with researchers and engineers to enhance model performance. Your analytical mindset enables you to develop cutting-edge training and profiling tools that contribute to the efficiency and reliability of language understanding and generation capabilities. You are passionate about driving innovation and are committed to the mission of scaling intelligence to serve humanity.

What you'll do

As a Performance Engineer in the Pre-Training team, you will be responsible for optimizing the performance of our advanced language models. Your primary focus will be on improving key model training metrics, ensuring high accelerator utilization, and developing robust systems. You will collaborate with cross-functional teams to identify performance bottlenecks and implement solutions that enhance the overall efficiency of our models. Your role will involve designing and developing training and profiling tools that support our mission of providing efficient language capabilities.

You will also engage in continuous improvement efforts, analyzing performance data to inform your strategies and decisions. By leveraging your expertise in software engineering and machine learning, you will contribute to the development of innovative solutions that push the boundaries of AI technology. Your work will have a direct impact on the capabilities of our models and the value they deliver to our customers.

What we offer

Cohere provides a supportive and inclusive work environment where you can thrive. We offer a range of benefits, including a budget for mental health support, generous parental leave, and personal enrichment benefits. Our flexible remote work policy allows you to balance your professional and personal life effectively. You will have the opportunity to work alongside some of the best minds in the industry, contributing to projects that shape the future of AI. Join us in our mission to scale intelligence and make a meaningful impact on the world.