Principal Model Optimization Engineer
RobloxUser-Generated Content company
San Mateo, United States$295,250 - $345,040 USDLead
Altimeter Capital
Dragoneer Investment Group
Andreessen Horowitz
Tencent
First Round Capital
Warner Music Group
Data & AINew
About the role
TL;DR
Optimize machine learning models for performance on GPU architectures.
- •Roblox's ML Platform supports hundreds of ML use cases and billions of inferences daily.
- •As a Model Optimization Engineer, you will optimize machine learning models for performance on GPU architectures, focusing on both training and inference workflows.
- •Key Responsibilities Optimize machine learning models for performance on GPU architectures.
- •Conduct low-level performance profiling analysis to identify bottlenecks.
- •Contribute to the development of best practices and tooling for model optimization.
- •Collaborate with cross-functional teams to integrate and deploy optimized models.
- •Requirements 6+ years of professional experience and system design experience.
- •Significant experience debugging GPUs (reading GPU profiles, debugging Xid errors).
- •Proficient in advanced tools and frameworks (e.g., CUDA, Triton, TensorRT).
- •Experience with model optimization techniques for LLMs.
Required skills
PythonLLMsComputer VisionNLP
Domain expertise
gamingdeveloper-toolsai
Benefits & perks
equity compensation, benefits
Tech stack
PythonLLMs