Machine Learning Engineer, Datasets

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