Member of Technical Staff, ML Optimization

2024-08-01
USA
Runway ML
Runway is a research company pioneering new tools for human imagination. Runway has been at the forefront of multi-modal AI systems ensuring that the future of content creation is accessible, controllable and empowering for creatives. Runway’s mission is to ensure that anyone anywhere can tell their stories. We believe that deep learning techniques applied to audiovisual content will forever change art, creativity, and design tools.
Runway is leading a shift to generative media that is unlocking an unprecedented level of creative potential. The invention of the camera 200 years ago forever changed our world – AI is a new kind of camera that will reshape storytelling forever and lead to full feature films that are entirely generated.About the role
We’re looking for a Senior ML Infrastructure Engineer to help us optimize our AI models for media generation. The ideal candidate for this role has extensive experience improving the performance of model training and inference, strong low-level understanding of GPU workloads, and thrives in fast-paced, high-ownership environment.
What you’ll do

Optimize our state-of-the-art AI models for video generation, such as Gen-1 and Gen-2
Build tooling to improve the efficiency and reliability of distributed training runs on Runway’s HPC cluster

What you’ll need

3+ years of experience in a role optimizing machine learning model inference and training on NVIDIA hardware
Knowledge of Python, C/C++, CUDA, and extensive experience profiling GPU performance and distributed training runs
Contributions to an ML framework (such as PyTorch), optimized runtimes for inference (such as TensorRT) or compilers (such as GCC)
Strong communication, collaboration, and documentation skills

Runway strives to recruit and retain exceptional talent from diverse backgrounds while ensuring pay equity for our team. Our salary ranges are based on competitive market rates for our size, stage and industry, and salary is just one part of the overall compensation package we provide.
There are many factors that go into salary determinations, including relevant experience, skill level and qualifications assessed during the interview process, and maintaining internal equity with peers on the team. The range shared below is a general expectation for the function as posted, but we are also open to considering candidates who may be more or less experienced than outlined in the job description. In this case, we will communicate any updates in the expected salary range.
Lastly, the provided range is the expected salary for candidates in the U.S. Outside of those regions, there may be a change in the range, which again, will be communicated to candidates.
Salary range: $300,000-$370,000Working at Runway
Great things come from great teams. We’d love to hear from you.
We’re committed to creating a space where our employees can bring their full selves to work and have equal opportunity to succeed. So regardless of race, gender identity or expression, sexual orientation, religion, origin, ability, age, veteran status, if joining this mission speaks to you, we encourage you to apply.
More about Runway

Runway Research
Runway's AI Film Festival
Creative Dialogues Series
Runway Studios
Our Behaviors and Company Mission
We raised a $141 million extension to our Series C

We're excited to be recognized as a best place to work Crain's | InHerSight | BuiltIn NYC | INC