PhD ML Engineer Intern: Knowledge Graph & NLP

TikTok

San Jose, Northern (CA, KY)

Hybrid

USD 68,000 - 97,000

Full time

6 hours ago
Be an early applicant
Application generator

Don’t send a generic resume — generate a resume and cover letter tailored to this exact role.

Get past ATS filters

Benefits offered by this job

Health insurance
Life insurance
Wellbeing benefits
Housing allowance
10 paid holidays per year

Job summary

TikTok is seeking PhD internship candidates for the Knowledge Graph team in San Jose to contribute to neural network research for e-commerce. You will help categorize products, influencers, merchants, and mine product information to support recommendations, while leading graph- and tag-related tasks.

Qualifying candidates are PhD students in related fields with experience in LLM/NLP, graph embeddings, ML, and Python/PyTorch/TensorFlow.

Qualifications

  • Currently pursuing a PhD in Software Development, Computer Science, Computer Engineering or related discipline.
  • Experience in LLM, NLP, NER, MT, QA, knowledge graph, information extraction, text mining or classification.
  • Computer vision, facial recognition, multimodal learning, object detection/tracking, image classification.
  • Machine learning topics including transfer learning, graph embedding, RL, recommendation.
  • Strong coding in Python/Java/C/C++, Spark; familiar with PyTorch or TensorFlow/Keras.

Responsibilities

  • Craft intricate knowledge graphs for product/content insights and brand/SPU development.
  • Architect knowledge maps detailing buyer and product connections.
  • Collaborate with cross-functional teams on research and development.

Skills

PhD in CS/Software
LLM
NLP
NER
Machine translation
Knowledge graph
Classification
Python/Java/C/C++
PyTorch
TensorFlow
Spark
Hadoop
Flask/Django

Education

PhD in Software Development / CS

Tools

Spark
Hadoop
Hive
Flink
Python
PyTorch
TensorFlow

Job description

TikTok is seeking PhD internship candidates for the Knowledge Graph team in San Jose to contribute to neural network research for e-commerce. You will help categorize products, influencers, merchants, and mine product information to support recommendations, while leading graph- and tag-related tasks.

Qualifying candidates are PhD students in related fields with experience in LLM/NLP, graph embeddings, ML, and Python/PyTorch/TensorFlow.

Get your free, confidential resume review.
or drag and drop your file here.
Similar jobs

Similar jobs worth comparing

PhD ML Engineer Intern, E-Commerce Knowledge Graph
PhD ML Engineer Intern, E-Commerce Knowledge Graph

TikTok • Seattle (WA)

On-site
USD 69,000 - 91,000
Health insurance
10 paid holidays per year
Paid sick time (56 hours or 40 hours)
+1
Knowledge Graph ML Engineer
Knowledge Graph ML Engineer

TikTok • Seattle (WA)

On-site
USD 154,000 - 301,000
Medical, dental, and vision insurance
401(k) with company match
Parental leave
+6
Knowledge Graph ML Engineer – E‑commerce & NLP
Knowledge Graph ML Engineer – E‑commerce & NLP

TikTok • Seattle (WA)

On-site
USD 154,000 - 301,000
Medical, Dental, Vision Insurance
401(k) with company match
Paid parental leave
+1
Knowledge Graph ML Engineer for Ecommerce
Knowledge Graph ML Engineer for Ecommerce

TikTok • San Jose (CA)

On-site
USD 162,000 - 388,000
Knowledge Graph ML Engineer – E-commerce & Recommendations
Knowledge Graph ML Engineer – E-commerce & Recommendations

TikTok • San Jose (CA)

Hybrid
USD 128,000 - 317,000
Medical Insurance
Dental & Vision Insurance
401(k) with company match
+7
PhD ML Scientist Intern - Recommendation Systems
PhD ML Scientist Intern - Recommendation Systems

TikTok • San Jose (CA)

On-site
USD 68,000 - 97,000
Health insurance
10 paid holidays
Housing allowance
PhD ML Engineer Intern — Search & Relevance
PhD ML Engineer Intern — Search & Relevance

TikTok • San Jose (CA)

On-site
Health insurance
Paid holidays
Housing allowances
PhD ML Engineer Intern: Search & Ranking Systems
PhD ML Engineer Intern: Search & Ranking Systems

TikTok • San Jose (CA)

Hybrid
USD 50,000 - 83,000
Health insurance from day one
Housing allowance (possible)
10 paid holidays per year
+1
PhD Research Scientist Intern — GenAI & Multimodal ML
PhD Research Scientist Intern — GenAI & Multimodal ML

TikTok • San Jose (CA)

On-site
USD 85,000 - 121,000
Health insurance
Housing allowance
Paid holidays
ML Engineer Intern – Search & Relevance (PhD)
ML Engineer Intern – Search & Relevance (PhD)

TikTok • San Jose (CA), Northern (KY)

Hybrid
USD 68,000 - 97,000
Health insurance
Life insurance
Wellbeing benefits
+2