Member of Technical Staff, GPU Optimization

Runway; United States; Remote; USD 60,000 - 80,000; Full time

Job summary

A global AI research and technology company is seeking a Member of Technical Staff specializing in GPU Optimization. The role involves developing innovative research projects in computer vision and collaborating closely with product teams to implement machine learning models. Candidates must have over 5 years of experience in machine learning and strong programming skills. This is a remote opportunity that values diversity and equal opportunity for all applicants.

Qualifications

• 5+ years of relevant engineering or research experience in machine learning, computer vision and/or graphics.
• Experience with CUDA, C++, and systems-level performance optimizations.
• Solid knowledge of at least one machine learning framework (e.g. PyTorch, TensorFlow).
• Strong programming skills and ability to write clean and maintainable research code.

Responsibilities

• Develop innovative research projects in computer vision, focusing on generative models for image and video.
• Collaborate closely with the product organization to bring machine learning models to production.

Skills

Machine learning

Computer vision

CUDA

C++

PyTorch

TensorFlow

Rapid prototyping

Strong communication

Job description

Member of Technical Staff, GPU Optimization

Remote

We are a global AI research and technology company focused on building Universal Simulation systems. The research we are doing and the tools we are building are maturing rapidly and are quickly becoming the foundation for how we will all soon approach making anything.

Our team consists of creative, open minded, caring and ambitious people who are determined to change the world. We aspire to continuously build impossible things and our ability to do so relies on building an incredible team.

About the role

We’re looking for Machine Learning Engineers who can help us build new multimodal AI systems for media generation. The ideal candidate for this role has a combination of strong engineering skills, extensive experience working on machine learning projects, and thrives in fast-paced, high-ownership environment.

What you’ll do

• Develop innovative research projects in computer vision, focusing on generative models for image and video.
• Collaborate closely with the rest of the product organization to bring cutting-edge machine learning models to production.

What you’ll need

• 5+ years of relevant engineering or research experience in machine learning, computer vision and/or graphics
• Experience with CUDA, C++ and systems level performance optimizations
• Solid knowledge of at least one machine learning framework (e.g. PyTorch, Tensorflow)
• Very strong programming skills and ability to write clean and maintainable research 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 communication, collaboration, and documentation skills

We’re committed to creating a space where our employees can bring their full selves to work and have equal opportunity to succeed. We are an equal opportunity employer and welcome applicants from diverse backgrounds.