PhD Machine Learning Engineer: Graph & Risk Innovation

TikTok; Singapore; On-site; SGD 120,000 - 180,000; Full time

Job summary

TikTok is seeking a highly skilled researcher/engineer to join our team focused on advanced AI risk detection for our e-commerce platform. You will mine and predict risks, build scalable graph learning systems, and apply ML to improve content and product experiences, working with multi-disciplinary teams to deliver impactful solutions.

You will contribute to cutting-edge ML research, explore graph learning and sequence modeling, and help scale models for production while collaborating with

Qualifications

• PhD completed or near completion in a technical field related to software/CS.
• Strong coding skills with deep learning/graph ML frameworks (PyTorch, TensorFlow, DGL, pyg, sklearn).
• Familiar with graph/sequence learning algorithms with depth and breadth.
• Experience with AI-Agent projects including agent workflow design or LLM-based apps.
• Solid ML foundation and industry experience is a plus.

Responsibilities

• Apply algorithms to mine, identify, and predict risks and low-quality merchants, products, and creators in e-commerce.
• Build data models to improve product experience and system efficiency.
• Develop large-scale graphs and graph learning platforms for community governance and relationships.
• Research cutting-edge ML/graph/sequence learning technologies and deploy scalable models.
• Collaborate with teams to apply ML to live e-commerce scenarios.

Skills

Deep learning

Graph neural networks

PyTorch

TensorFlow

DGL

Python programming

LLM-based applications

Multi-agent systems

Education

PhD in Computer Science/Software Engineering/Related

Tools

Hive

Spark

Hadoop

NumPy

Pandas

Job description

TikTok is seeking a highly skilled researcher/engineer to join our team focused on advanced AI risk detection for our e-commerce platform. You will mine and predict risks, build scalable graph learning systems, and apply ML to improve content and product experiences, working with multi-disciplinary teams to deliver impactful solutions.

You will contribute to cutting-edge ML research, explore graph learning and sequence modeling, and help scale models for production while collaborating with