
Senior/Principal AI Platform Engineer - Feature Store Infrastructure
See more jobs from Roblox Corporation3 months old
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.
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 Knowledge Graphs and Feature Store Infrastructure, you will design and build next-generation ML data infrastructure, 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 Knowledge Graphs 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:
- Build a groundbreaking Knowledge Graph: Develop large-scale graph-based infrastructure that connects millions of players, thousands of developers, and billions of assets, unlocking AI-powered recommendations, safety, social insights, and fraud detection.
- 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.
- 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.
- 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, real-time AI serving, and Ground Truth Data Labeling.
You have:
- 5+ years of experience in AI/ML data infrastructure, knowledge graphs, feature stores.
- 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.
Why Roblox?
- Be a pioneer: Shape the AI landscape at a company redefining online experiences.
- Think big: Work on systems with global reach and immense scale.
- Stay at the cutting edge: Leverage the latest tools and technologies in AI/ML infrastructure.
- Change the game: Your work will set the standard for what’s possible in AI at scale.
Take the leap and help us redefine the future of AI at Roblox. Together, we’ll create experiences that connect billions and inspire the next generation of creators.
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.
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.