AI Data Engineer Jobs at Roblox with Visa Sponsorship

INTRODUCTION

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.

ROLE AND RESPONSIBILITIES

Power the Future of AI at Roblox!

Join the AI Platform team at Roblox, where we are building the infrastructure that enables world-class AI/ML experiences. Our mission is to empower developers and creators by providing scalable, high-performance systems that drive AI innovation at an unprecedented scale.

As an AI Platform Engineer specializing in ML Infrastructure, Training data Infrastructure and Feature Store Infrastructure, you will design and build next-generation ML data platforms, enabling AI applications across recommendations, search, safety, content generation, fraud detection, and more. Your work will directly impact millions of users, supporting AI-driven experiences across the entire Roblox ecosystem.

Why This Role?

• Massive Impact: Your contributions will support billions of daily interactions, powering AI-driven experiences across gaming, learning, social connections, and digital creativity.
• End-to-End Ownership: Be a key architect in defining and building Roblox’s Training Data pipelines and Feature Store infrastructure from the ground up.
• Cutting-Edge Growth: Work at the intersection of AI/ML, data engineering, and distributed systems, solving some of the most complex large-scale challenges in the industry.

You will:

• Develop a scalable Feature Store: Architect and implement a system that supports both batch and real-time feature computation, storage, and serving, enabling low-latency AI inference at scale. Develop Sequence Feature Stores to collect raw user events and power training of transform models for recommendation engines at Roblox.
• Develop Roblox’s first training data platform: Architecture and engineer training data collection, reusability and reliability while solving world class engineering problems like point-in-time joins, feature injection. Power batch and realtime model training.
• Engineer large-scale ML data pipelines: Design, optimize, and maintain high-throughput streaming and batch ETL pipelines that ingest, process, and serve billions of records per day.
• Build a feature engineering framework to empower 100s of machine learning engineers to develop, author, experiment and productionize millions of signals into the system quickly, easily and reliably.
• Enable online and offline inference: Ensure real-time, low-latency feature access for AI inference while maintaining consistency and robustness for batch model training.
• Collaborate with cross-functional AI teams: Partner with ML researchers, data engineers, and product teams to build the next-gen AI infrastructure that fuels innovation at Roblox.
• Research & innovate: Stay at the forefront of AI infrastructure by exploring the latest in VectorDBs, Feature Stores, Graph ML, Agentic Systems and Real-time AI serving.

BASIC QUALIFICATIONS

• 5+ years of experience in AI/ML data infrastructure, feature stores, training data infrastructure. Experience working with industry leading NoSql databases like DynamoDB, Cassandra, cockroachDB and many more.
• Expertise in building scalable ML data pipelines for both batch and real-time environments.
• Strong software engineering skills with experience in distributed computing frameworks (e.g., Spark, Flink, Ray) and real-time data systems (e.g., Kafka, Redis, DynamoDB).
• Deep knowledge of ML Feature Stores (e.g., Feast, Tecton, Vertex AI Feature Store) and embedding/vector search infrastructure.
• Familiarity with Knowledge Graphs and graph-based ML applications (e.g., Neo4j, TigerGraph, AWS Neptune).
• A Bachelor's degree in Computer Science, Engineering, or a related field.

COMPENSATION

• Annual Salary Range: $277,170—$343,340 USD

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.

Roles that are based in an office are onsite Tuesday, Wednesday, and Thursday, with optional presence 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 to candidates with qualifying disabilities or religious beliefs during the recruiting process.