Staff ML Engineer, Optimization

August 30, 2023

Apply Now

Get a free AI resume review

Runway

WebsiteLinkedInAll Job Openings

Next-generation content creation with artificial intelligence

machine learning • artificial intelligence • tools • creative • Computer Software

11 - 50

💰 $50M Series C on 2022-12

Description

• 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

Requirements

• 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

Benefits

• Competitive salary • Equal opportunity • Diverse and inclusive work environment