Principal Model Optimization Engineer

Required

• 6+ years of professional experience and a tool chest of system design experience upon which to draw to build performant systems for all of Roblox.
• A track record of building ML infrastructure end to end. Examples include training pipelines, workflow orchestration (e.g. Kubeflow, Airflow, Ray), data or feature pipelines, experiment tracking and model registries, or model serving.
• Experience designing SDKs, APIs or developer tools that were widely adopted, and a strong sense of what makes a platform easy to use.
• Hands-on experience running GPU workloads at scale on Kubernetes. That includes scheduling, quotas, multi-cluster dispatch, autoscaling, and debugging GPU and node failures (e.g. Xid errors, NCCL hangs).
• Enough depth in training and inference performance (distributed training, mixed precision, batching, quantization) to know where the platform should do the work for its users.
• A habit of measuring: you set clear success metrics, ship in steps, and keep iterating until the numbers move.
• Experience leading cross-team technical work without direct authority, and writing clearly for both engineers and leaders.
• A Bachelor’s degree in Computer Science, Computer Engineering or a related field, or equivalent experience.
• For roles that are based at our headquarters in San Mateo, CA: The starting base pay for this position is as shown below. The actual base pay is dependent upon a variety of job-related factors such as professional background, training, work experience, location, business needs and market demand. Therefore, in some circumstances, the actual salary could fall outside of this expected range. This pay range is subject to change and may be modified in the future. All full-time employees are also eligible for equity compensation and for benefits as described on this page.
• Annual Salary Range
• $295,250—$345,040 USD
• Roles that are based in an office are onsite Tuesday, Wednesday, and Thursday, with optional presence on Monday and Friday (unless otherwise noted).