Runway is hiring a
Member of Technical Staff, ML Optimization in Worldwide

Summary

Runway is seeking a Senior ML Infrastructure Engineer to optimize their AI models for video generation. The ideal candidate has 3+ years of experience in optimizing machine learning model inference and training on NVIDIA hardware, knowledge of Python, C/C++, CUDA, and contributions to an ML framework or compilers.

Requirements

• 3+ years of experience in a role optimizing machine learning model inference and training on NVIDIA hardware
• Knowledge of Python, C/C++, CUDA
• Contributions to an ML framework (such as PyTorch), optimized runtimes for inference (such as TensorRT) or compilers (such as GCC)

Responsibilities

• 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

Preferred Qualifications

Strong communication, collaboration, and documentation skills

Benefits

Salary range: $300,000-$370,000