Software Engineering • Mid-Level

Who you are

You have at least 5 years of experience in software engineering, particularly in optimizing machine learning models for production environments. You possess a strong understanding of modern ML architectures and are adept at enhancing their performance, especially for inference tasks. Your problem-solving skills are top-notch, and you are willing to learn and adapt to tackle challenges effectively. You thrive in collaborative environments, working alongside researchers and engineers to bring cutting-edge technologies into production. You have experience with cloud platforms, particularly Azure, and understand how to leverage them for optimal performance.

Desirable

Experience with high-volume, low-latency systems is a plus, as is familiarity with tools that provide visibility into system performance and bottlenecks. You are comfortable working in a fast-paced environment and are eager to contribute to innovative solutions that push the boundaries of AI technology.

What you'll do

In this role, you will collaborate with machine learning researchers and product managers to optimize OpenAI's advanced AI models for production use. You will introduce new techniques and tools to improve the performance, latency, and efficiency of the model inference stack. Your responsibilities will include building tools to identify bottlenecks and sources of instability, then designing and implementing solutions to address these issues. You will optimize code and manage a fleet of Azure VMs to maximize hardware utilization, ensuring that every FLOP and GB of GPU RAM is effectively used. You will also contribute to the overall research progression by enabling advanced research through robust engineering practices.

What we offer

At OpenAI, you will be part of a mission-driven team that believes in the transformative potential of artificial intelligence. We offer a collaborative work environment where your contributions will directly impact the future of technology. You will have opportunities for professional growth and development, working with some of the brightest minds in the field. We are committed to providing reasonable accommodations to applicants with disabilities, ensuring an inclusive workplace for all. Join us in shaping the future of AI and making a difference in the world.