Ai Engineer • Mid-Level

Who you are

You have a strong background in machine learning and model optimization, with experience in improving model efficiency in production environments. You are proficient in Python and have hands-on experience with frameworks like TensorFlow and PyTorch, enabling you to dive deep into model execution and identify bottlenecks effectively.

You thrive in collaborative environments, working closely with researchers and engineers to experiment, measure, and implement improvements that enhance inference performance. Your analytical mindset allows you to develop innovative optimizations that drive lower latency and higher throughput across diverse workloads.

What you'll do

As a Member of Technical Staff, you will be responsible for building reliable machine learning systems that push the boundaries of LLM inference efficiency. You will collaborate with modeling and systems teams to experiment with various techniques and measure their impact on core performance metrics. Your role will involve diving deep into model execution, identifying bottlenecks, and developing optimizations that enhance the overall performance of AI models in production.

You will contribute to the team's mission of scaling intelligence to serve humanity by ensuring that the models we deploy are efficient and effective. Your work will directly impact the capabilities of our models and the value they provide to customers, making you an integral part of our mission to drive the widespread adoption of AI.

What we offer

Cohere provides a supportive work environment with a focus on mental health and personal enrichment. We offer a generous parental leave policy, flexible remote work options, and a budget for personal development in areas such as arts, culture, and fitness. You will enjoy six weeks of vacation, allowing you to recharge and maintain a healthy work-life balance. Join us in shaping the future of AI and making a meaningful impact in the industry.