Senior/Principal AI Platform Engineer jobs in United States
cer-icon
Apply on Employer Site
company-logo

Roblox · 1 day ago

Senior/Principal AI Platform Engineer

Roblox is a platform empowering users to create and connect through immersive digital experiences. The AI Platform Engineer will design and build the infrastructure for AI/ML experiences, impacting millions by developing scalable systems for data processing and AI applications.

3D TechnologyGamingMetaverseOnline GamesSoftwareVideo Games
check
Comp. & Benefits
check
H1B Sponsor Likelynote

Responsibilities

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

Qualification

AI/ML data infrastructureFeature Store infrastructureML data pipelinesDistributed computing frameworksNoSQL databasesReal-time data systemsML Feature StoresKnowledge GraphsResearch & innovationSoftware engineeringCollaborationProblem-solving

Required

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

Benefits

Equity compensation
Benefits as described on this page

Company

Roblox is an online gaming and entertainment platform that offers a shared digital experience that brings people together through play.

H1B Sponsorship

Roblox has a track record of offering H1B sponsorships. Please note that this does not guarantee sponsorship for this specific role. Below presents additional info for your reference. (Data Powered by US Department of Labor)
Distribution of Different Job Fields Receiving Sponsorship
Represents job field similar to this job
Trends of Total Sponsorships
2025 (240)
2024 (111)
2023 (139)
2022 (153)
2021 (91)
2020 (92)

Funding

Current Stage
Public Company
Total Funding
$874.45M
Key Investors
Andreessen HorowitzAltos Ventures
2024-10-01Post Ipo Equity
2022-04-12Series Unknown· $17.71M
2021-08-11Series Unknown

Leadership Team

leader-logo
David Baszucki
CEO & Founder
linkedin
leader-logo
Michael Guthrie
CFO
linkedin
Company data provided by crunchbase