Senior​/Principal AI Platform Engineer - Feature Store

Base pay range

$/yr - $/yr

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.

• 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 Vector
  
  DBs, Feature Stores, Graph ML, real-time AI serving, and Ground Truth Data Labeling.

• 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, Dynamo
  
  DB).
• 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, Tiger Graph, AWS Neptune).
• A Bachelor's degree in Computer Science, Engineering, or a related field.

• 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…