Senior/Principal AI Platform Engineer

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.