Runway AI is looking for Machine Learning Engineers to help build new multimodal AI systems for media generation, aiming to make the creation of various media types (images, videos, games, etc.) accessible to everyone without barriers.
Requirements
- 5 years of experience in a role optimizing machine learning model inference and training
- 5 years of experience with Python, C/C++, CUDA
- 5 years of experience profiling GPU performance and distributed training runs
- Experience with ML framework (such as PyTorch), optimized runtimes for inference (such as TensorRT) or compilers (such as GCC)
Responsibilities
- Improve performance of model training and inference.
- Exercise strong low-level understanding of GPU workloads in a fast-paced, high-ownership environment.
- Optimize state-of-the-art AI models for video generation, such as Gen-3 and other video and video models.
- Build tooling to improve the efficiency and reliability of distributed training runs on Runway’s HPC cluster.
Other
- Open to hiring remote across America — we also have offices in NYC, San Francisco and Seattle*
- Role may telecommute from anywhere in the US with occasional travel to HQ.
- 5 years of experience in the job offered or similar software engineering position
- Bachelor’s degree in Computer Science, Mathematics, Applied Mathematics, or a closely related field, or foreign degree equivalent
- Apply directly through this website by submitting your resume and cover letter below.
