AI Infrastructure Engineer — LLM & GPU Systems

TikTok; Singapore; On-site; SGD 65,000 - 90,000; Full time

Job summary

TikTok is hiring for a New Graduate role in Model Infrastructure to help build the next generation of AI infrastructure for TikTok's recommendation systems and LLMs.

You will work on distributed training, high-performance GPU computing, and production-ready AI systems, collaborating with researchers to deploy training and serving solutions. Applicants should be recent graduates with a passion for large-scale AI systems.

Qualifications

• Bachelor's or Master's degree in AI, software development, CS, or related discipline.
• Strong programming skills in C++ or Python.
• Good understanding of data structures, algorithms, and computer systems.
• Familiarity with PyTorch or TensorFlow.
• Knowledge of Transformer architectures and Large Language Models (LLMs).
• Strong problem-solving skills and a passion for building large-scale AI systems.

Responsibilities

• Build and optimize infrastructure for large-scale model training and online inference.
• Develop distributed systems supporting large recommendation models and LLMs.
• Improve training and inference performance through GPU optimization and efficient communication.
• Collaborate with researchers to develop and deploy LLM training and serving solutions.
• Analyze system bottlenecks and implement performance optimizations.

Skills

Bachelor's degree in AI/CS/Engineering

C++ or Python

Data structures/algorithms

PyTorch or TensorFlow

Transformer/LLMs knowledge

Problem solving

Education

Bachelor's or Master's degree in AI/Software/CS/Engineering

Tools

CUDA

Triton

Job description

TikTok is hiring for a New Graduate role in Model Infrastructure to help build the next generation of AI infrastructure for TikTok's recommendation systems and LLMs.

You will work on distributed training, high-performance GPU computing, and production-ready AI systems, collaborating with researchers to deploy training and serving solutions. Applicants should be recent graduates with a passion for large-scale AI systems.