Senior / Principal Model Optimization Engineer

1 Month ago • 4 Years + • Software Development & Engineering • $233,840 PA - $283,780 PA

Job Summary

Job Description

Roblox is seeking experienced Machine Learning Engineers for its ML Platform team to optimize model performance for training and inference on GPU architectures. Responsibilities include low-level performance profiling, identifying bottlenecks, developing best practices for optimization and deployment, and collaborating with data scientists and software engineers. The role requires deep expertise in debugging GPUs and experience with model optimization techniques for LLMs such as speculative decoding, continuous batching, and quantization.
Must have:
  • 4+ years of professional experience
  • System design experience
  • GPU debugging experience
  • Bachelor's degree in a technical field
Good to have:
  • Experience with LLM optimization techniques
  • Proficiency in CUDA, Triton, TensorRT
  • Passion for performance optimization
  • Advocacy for generalization and tool building
  • Support for internal partners

Job Details

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 machine learning 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 as described on this page.

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).

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.

Similar Jobs

DraftKings - Senior iOS Engineer

DraftKings

Bulgaria (Remote)
1 Month ago
Ubisoft - Senior ML Data Scientist

Ubisoft

Montreal, Quebec, Canada (On-Site)
3 Months ago
FICO - Scores Credit Bureau Alliances Relationship Manager

FICO

United States (Remote)
5 Days ago
Aristocrat - Field Engineer I

Aristocrat

Oklahoma City, Oklahoma, United States (On-Site)
1 Month ago
Yodlee - Investments Marketing Manager

Yodlee

Berwyn, Pennsylvania, United States (Hybrid)
1 Month ago
Crowd Strick - Engineering Manager, Falcon Foundry

Crowd Strick

United Kingdom (Remote)
3 Days ago
Autodesk - Senior Director, Cloud Infrastructure & Platforms Engineering (CIPE)

Autodesk

San Francisco, California, United States (On-Site)
1 Month ago
Apple - AR/VR Software Development Engineer

Apple

Cupertino, California, United States (On-Site)
2 Months ago
entrata - Sales Engineer

entrata

United States (Remote)
1 Month ago
Notion - Software Engineer, Search

Notion

New York, New York, United States (On-Site)
9 Months ago

Get notifed when new similar jobs are uploaded

Similar Skill Jobs

Mapbox - Technical Program Manager III

Mapbox

United Kingdom (Remote)
2 Weeks ago
Moloco - Strategy & Operations Manager

Moloco

Beijing, China (On-Site)
1 Month ago
GoTo Group - Product Designer

GoTo Group

Jakarta, Indonesia (On-Site)
1 Month ago
Enphase Energy - Customer Service Manager

Enphase Energy

Bengaluru, Karnataka, India (On-Site)
3 Weeks ago
bytro studios - UI/UX Artist

bytro studios

Hamburg, Hamburg, Germany (Hybrid)
1 Month ago
Niantic - Senior UX Designer

Niantic

San Francisco, California, United States (Hybrid)
4 Months ago
Krafton India  - Lead Marketing Manager

Krafton India

Bengaluru, Karnataka, India (On-Site)
1 Month ago
Brillio - Data Architect

Brillio

Plainsboro Township, New Jersey, United States (Hybrid)
2 Months ago
Plaid  - Engineering Manager - Credit Dashboard

Plaid

San Francisco, California, United States (On-Site)
1 Month ago
Grab - Senior Salesforce Business Analyst

Grab

Bengaluru, Karnataka, India (On-Site)
1 Month ago

Get notifed when new similar jobs are uploaded

Jobs in San Mateo, California, United States

ChainGuard - Strategic Business Development Representative

ChainGuard

United States (Remote)
3 Days ago
Nova Credit - Staff Product Designer

Nova Credit

New York, United States (Remote)
2 Months ago
Riot Games - Senior Game Producer

Riot Games

Los Angeles, California, United States (On-Site)
2 Months ago
Lambda - Senior HPC Systems Architect

Lambda

San Jose, California, United States (Hybrid)
2 Weeks ago
Reltio - Senior Marketing Analytics Manager

Reltio

North Carolina, United States (On-Site)
2 Days ago
illumio - Staff Engineer - Architecture Team

illumio

Sunnyvale, California, United States (On-Site)
3 Days ago
Optiv - Sr. SOAR Engineer - Phantom

Optiv

Kansas City, Missouri, United States (Remote)
2 Months ago
PayPal - Sr Staff Software Engineer - Android

PayPal

San Jose, California, United States (Hybrid)
1 Month ago
Whalar - Summer Intern

Whalar

New York, United States (On-Site)
1 Month ago
Nintendo - CONTRACT - Marketing Translation Coordinator (French)

Nintendo

Redmond, Washington, United States (Hybrid)
7 Months ago

Get notifed when new similar jobs are uploaded

Software Development & Engineering Jobs

Valeo - Methods Engineer Intern

Valeo

Zaragoza, Aragon, Spain (On-Site)
7 Months ago
BigID - Senior Software Engineer

BigID

Buenos Aires, Buenos Aires, Argentina (Remote)
1 Week ago
Neolytix - Healthcare Integration Engineer

Neolytix

Gurugram, Haryana, India (On-Site)
2 Weeks ago
Google - Technical Lead Software Engineering, Chrome Enterprise Premium

Google

Warsaw, Masovian Voivodeship, Poland (On-Site)
2 Months ago
Guardian - Lead Engineer - IAM

Guardian

Chennai, Tamil Nadu, India (Hybrid)
1 Day ago
Coda - Senior/Staff Software Engineer

Coda

Bangkok, Thailand (Hybrid)
1 Year ago
Nagarro - Staff Engineer, Sharepoint

Nagarro

India (Remote)
9 Months ago
Coda - Software Engineering Manager

Coda

Singapore (Hybrid)
1 Week ago
Toast - Software Engineering Manager - Fintech, Payments Reporting & Data

Toast

Dublin, County Dublin, Ireland (Hybrid)
2 Weeks ago
Assystems - Sr. Structural Design Engineer - Buildings

Assystems

Hyderabad, Telangana, India (On-Site)
8 Months ago

Get notifed when new similar jobs are uploaded

About The Company

San Mateo, California, United States (On-Site)

San Mateo, California, United States (On-Site)

Washington, District Of Columbia, United States (Hybrid)

San Mateo, California, United States (On-Site)

Gurugram, Haryana, India (Hybrid)

San Mateo, California, United States (On-Site)

San Mateo, California, United States (On-Site)

San Mateo, California, United States (On-Site)

San Mateo, California, United States (Hybrid)

View All Jobs

Get notified when new jobs are added by Roblox

Level Up Your Career in Game Development!

Transform Your Passion into Profession with Our Comprehensive Courses for Aspiring Game Developers.

Job Common Plug