Runway
Next-generation content creation with artificial intelligence
machine learning • artificial intelligence • tools • creative • Computer Software
11 - 50
Staff ML Engineer - Optimization
August 30, 2023
🏡 Remote – New York
Runway
Next-generation content creation with artificial intelligence
machine learning • artificial intelligence • tools • creative • Computer Software
11 - 50
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
Apply Now