Skip to content
Senior/Principal AI Platform Engineer – Knowledge Graphs & Feature Store Infrastructure
| Company | Roblox |
|---|
| Location | San Mateo, CA, USA |
|---|
| Salary | $238520 – $391590 |
|---|
| Type | Full-Time |
|---|
| Degrees | Bachelor’s |
|---|
| Experience Level | Senior |
|---|
Requirements
- 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.
Responsibilities
- 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.
Preferred Qualifications
No preferred qualifications provided.
