Senior / Principal Model Optimization Engineer

Posted Yesterday
Be an Early Applicant
San Mateo, CA
234K-284K Annually
Senior level
Computer Vision • Gaming • Software • Virtual Reality • Web3 • Metaverse
Immersive gaming and creation platform
The Role
As a Model Optimization Engineer at Roblox, you will optimize machine learning models for GPU architectures, conduct performance profiling, and collaborate with teams to integrate optimized models into production. You will also develop best practices and tooling to enhance model execution and work on improving overall performance.
Summary Generated by Built In

Every day, tens of millions of people come to Roblox to explore, create, play, learn, and connect with friends in 3D immersive digital experiences– all created by our global community of developers and creators. 

At Roblox, we’re building the tools and platform that empower our community to bring any experience that they can imagine to life. Our vision is to reimagine the way people come together, from anywhere in the world, and on any device. We’re on a mission to connect a billion people with optimism and civility, and looking for amazing talent to help us get there. 

A career at Roblox means you’ll be working to shape the future of human interaction, solving unique technical challenges at scale, and helping to create safer, more civil shared experiences for everyone.

ML Platform @ Roblox today supports hundreds of ML use cases and billions of inferences per day across Discovery, Safety, Engine, and much more. As a Model Optimization engineer on ML Platform, you will be responsible for digging deep into model internals to optimize performance, for both training and inference. We are looking for accomplished engineers to help us maximize performance of our platform.

You Will:

  • Optimize machine learning models for performance on GPU architectures, focusing on both training and inference workflows.
  • Conduct low-level performance profiling analysis to identify bottlenecks in existing machine learning pipelines and propose actionable improvements.
  • Contribute to the development of best practices and tooling for model optimization and deployment.
  • Collaborate with cross-functional teams, including data scientists and software engineers, to integrate and deploy optimized models into production environments.
  • Partner across organizations to build tooling, interfaces, and visualizations that make the ML@Roblox a delight to use.

You Have:

  • 4+ years of professional experience and a tool chest of system design experience upon which to draw to build performant systems for all of Roblox.
  • Significant experience debugging GPUs - reading GPU profiles, debugging Xid errors, etc.
  • Ideally have experience with model optimization techniques for LLMs, such as speculative decoding, continuous batching, quantization, etc.
  • Bachelor's degree in Computer Science, Computer Engineering, Data Science, or a similar technical field.

You Are:

  • Proficient in advanced tools and frameworks (e.g., CUDA, Triton, TensorRT) to enhance model execution speed and reduce latency.
  • A performance nut; you love pushing the limits of what’s possible, whether it’s squeezing every last ounce of efficiency from a GPU, fine-tuning algorithms for peak speed, or innovating new techniques to enhance model performance
  • A generalization advocate: you're passionate about building tools and frameworks that consistently deliver improvements in model performance.
  • Passionate about supporting internal partners (data scientists and ML Engineers) to meet and understand their needs.

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.

Annual Salary Range

$233,840$283,780 USD

Roles that are based in our San Mateo, CA Headquarters are in-office Tuesday, Wednesday, and Thursday, with optional in-office on Monday and Friday (unless otherwise noted).

You’ll Love: 

  • Industry-leading compensation package
  • Excellent medical, dental, and vision coverage
  • A rewarding 401k program
  • Flexible vacation policy (varies by exemption status)
  • Roflex - Flexible and supportive work policy 
  • Roblox Admin badge for your avatar
  • At Roblox HQ: 
    • Free catered lunches five times a week and several fully stocked kitchens with unlimited snacks
    • Onsite fitness center and fitness program credit
    • Annual CalTrain Go Pass

Roblox provides equal employment opportunities to all employees and applicants for employment and prohibits discrimination and harassment of any type without regard to race, color, religion, age, sex, national origin, disability status, genetics, protected veteran status, sexual orientation, gender identity or expression, or any other characteristic protected by federal, state or local laws. Roblox also provides reasonable accommodations for all candidates during the interview process.

Top Skills

Cuda
Tensorrt
Triton

What the Team is Saying

Claus
Ying
Denise
Andrea
The Company
HQ: San Mateo, CA
2,500 Employees
Hybrid Workplace
Year Founded: 2004

What We Do

Roblox is an immersive gaming and creation platform that offers people millions of ways to be together, inviting its community to explore, create and share endless unique experiences. Our vision is to reimagine the way people come together — in a world that's safe, civil, and optimistic. To achieve this vision, we are building an innovative company that, together with the Roblox community, has the ability to strengthen our social fabric and support economic growth for people around the world.

Why Work With Us

Every day, Roblox employees tackle complex questions and drive incredible innovation. Join us in building the future of human connection and communication.

Gallery

Gallery
Gallery
Gallery
Gallery

Roblox Offices

Hybrid Workspace

Employees engage in a combination of remote and on-site work.

We are requiring employees to be in the office three days a week – with core days being Tuesday, Wednesday, and Thursday. On Mondays and Fridays, employees may choose to work remotely, although the office will still be open.

Typical time on-site: 3 days a week
Company Office Image
HQSan Mateo, CA
IN
Learn more

Similar Jobs

Roblox Logo Roblox

Senior / Principal Infrastructure Engineer - ML Platform

Computer Vision • Gaming • Software • Virtual Reality • Web3 • Metaverse
San Mateo, CA, USA
2500 Employees
234K-284K Annually

Roblox Logo Roblox

Senior Software Engineer, Communication Safety

Computer Vision • Gaming • Software • Virtual Reality • Web3 • Metaverse
San Mateo, CA, USA
2500 Employees
189K-234K Annually

Roblox Logo Roblox

Principal Machine Learning Engineer, Critical Harms

Computer Vision • Gaming • Software • Virtual Reality • Web3 • Metaverse
San Mateo, CA, USA
2500 Employees
284K-332K Annually

Roblox Logo Roblox

Engineering Manager-Data Platform

Computer Vision • Gaming • Software • Virtual Reality • Web3 • Metaverse
San Mateo, CA, USA
2500 Employees
284K-332K Annually

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account