Runway AI - Member of Technical Staff, Machine Learning Datasets

AI Developer is needed in United States.

Client: Runway AI

Location: Remote

Contract: undefined

Job Description

We are building AI to simulate the world through merging art and science. We believe that world models are at the frontier of progress in artificial intelligence, requiring models that experience the world and learn from their mistakes in a simulated environment. Our team consists of creative, open-minded, caring, and ambitious individuals determined to change the world.

We are looking for Dataset Engineers to help curate, build, and optimize datasets for model training. The ideal candidate has strong machine learning skills and extensive experience with large-scale datasets and creativity tools.

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 the 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

Salary Information

Salary range: $270,000-$370,000. The salary is based on competitive market rates for our size, stage, and industry, considering relevant experience, skill level, and qualifications assessed during the interview process. The range provided is expected for candidates in the U.S. and may vary outside this region.

Working at Runway

Great things come from great teams. We are committed to creating a space where our employees can bring their full selves to work and have equal opportunity to succeed, regardless of race, gender identity or expression, sexual orientation, religion, origin, ability, age, or veteran status.