Graduate Data Platform & AI Knowledge Scientist

TikTok Inc.; San Jose (CA); On-site; USD 218,000 - 480,000; Full time

Benefits offered by this job

Medical, dental, and vision insurance

401(k) with company match

Paid parental leave

Short-term and long-term disability

Life insurance

Wellbeing benefits

10 paid holidays per year

10 paid sick days per year

17 days of Paid Personal Time

Job summary

TikTok Inc. is seeking a graduate to help build a unified knowledge layer for enterprise data, enabling LLMs to access and reason over structured and unstructured data across warehouses, logs, and streams.

You will work at the intersection of data infrastructure, AI systems, and knowledge modeling, shaping how AI interacts with enterprise data and supporting analytics, decision-making, and automation.

Qualifications

• PhD candidate or completed in software development, CS, CE or related field.
• Strong programming skills in Python/Java/Scala.
• Solid understanding of data systems, data modeling, or distributed systems.
• Experience in data engineering/backend systems or ML/LLM systems.

Responsibilities

• Research and design unified knowledge representations for enterprise data.
• Explore and build RAG-based knowledge systems with high accuracy and low latency.
• Develop ontology/semantic layers to bridge data and LLM understanding.
• Design knowledge ingestion and update mechanisms (batch + real-time).
• Improve LLM grounding, traceability, and reliability.
• Explore agent-based reasoning and execution frameworks.
• Prototype and validate new ideas, and bring them into production systems.

Skills

Python/Java/Scala

Data systems

ML/LLM systems

Education

PhD in Software Development/CS/CE

Tools

Flink

Kafka

Job description

TikTok Inc. is seeking a graduate to help build a unified knowledge layer for enterprise data, enabling LLMs to access and reason over structured and unstructured data across warehouses, logs, and streams.

You will work at the intersection of data infrastructure, AI systems, and knowledge modeling, shaping how AI interacts with enterprise data and supporting analytics, decision-making, and automation.