Machine Learning Engineer, Datasets

Runway; United States; Remote; USD 260,000 - 325,000; Full time

Job summary

An innovative firm is seeking Dataset Engineers to curate and optimize datasets for cutting-edge machine learning models. This role involves collaborating with a world-class research team to push the boundaries of content creation. Ideal candidates will have strong machine learning skills and experience with large-scale datasets. You'll be at the forefront of generative media, ensuring data quality and implementing fast iteration cycles. Join a company committed to creativity and diversity, where your contributions will help reshape storytelling for the future.

Qualifications

• 4+ years of experience in machine learning or dataset engineering.
• Proficiency in model finetuning and optimization for data preprocessing.
• Strong familiarity with machine learning frameworks.

Responsibilities

• Develop and maintain large-scale, multimodal datasets for training models.
• Optimize models for data preprocessing tasks.
• Evaluate new datasets and models for upstream data tasks.

Skills

Machine Learning

Data Analysis

SQL

Programming

Data Curation

Model Optimization

Creativity Tools

Tools

PyTorch

JAX

TensorFlow

Ray

Kubernetes

Airflow

Prefect

Job description

Runway is an applied research company pioneering new tools for human imagination. Runway has been at the forefront of multi-modal AI systems ensuring that the future of media 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.

About the role

*Open to hiring remote across North America and Europe — we also have offices in NYC, San Francisco, Seattle, and London

We're looking for Dataset Engineers to help curate, build, and optimize datasets for model training. The ideal candidate for this role has strong machine learning skills, extensive experience working with and analyzing large-scale datasets, and an understanding of creativity tools. You should be proficient in ensuring data quality and tight feedback loops between data preprocessing and model training.

What you'll do

• Develop and maintain large-scale, multimodal datasets for training and evaluating models
• Optimize models for data preprocessing tasks
• Create and run evaluations and benchmark analyses for datasets and models
• Implement fast iteration cycles and feedback loops to continuously improve model datasets
• Work with a world-class research team to push the boundaries of content creation
• Evaluate new datasets and models for upstream data tasks that feed into our products

What you'll need

• 4+ years of relevant experience in machine learning or dataset engineering, ideally with multimodal datasets
• Experience with running and optimizing models offline at large scale
• Excellent data modeling skills and experience with data curation
• Proficiency in model finetuning and optimization for data preprocessing
• Strong data analysis and SQL skills
• Experience in creating evaluations and running benchmark analyses
• Solid knowledge of at least one machine learning framework (e.g. PyTorch, JAX, TensorFlow)
• Very strong programming skills and ability to write clean and maintainable code
• Deep interest in building human-in-the-loop systems for creativity
• Ability to rapidly prototype solutions and iterate on them with tight product deadlines
• Strong familiarity with tools such as Ray, Kubernetes, Airflow, Prefect
• Excellent 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: $260,000-$325,000

Working 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're excited to be recognized as a best place to work Crain's|InHerSight|BuiltIn NYC | INC