Model Infra Engineer for Large-Scale Recommenders

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

Benefits offered by this job

Medical, dental, and vision insurance

401(k) with company match

Paid parental leave

Short/long-term disability coverage

Wellbeing benefits

Paid holidays and PTO

Job summary

TikTok Inc. in San Jose is seeking talented Software Engineers to join our Model Infrastructure team. You will focus on architecture and engineering challenges for large-scale recommendation models, from online inference to offline training, to support evolving foundation models.

You will co-design infrastructure with researchers to balance compute cost with recommendation effectiveness, optimize performance, and enable scalable experimentation across GPU clusters.

Qualifications

• Bachelor's degree or above in Computer Science or related fields.
• Understanding of GPU architecture and the software stack (CUDA, CUTLASS, Triton Lang), with experience in GPU performance analysis.
• Experience with at least one machine learning framework (e.g., TensorFlow, PyTorch, or in-house alternatives).
• Initiative and self-drive; eager to take on complex challenges.
• Proven ability to work independently and collaboratively in a fast-paced team environment.

Responsibilities

• Build and optimize infrastructure for online inference and offline training of recommendation models.
• Solve challenges posed by the combination of high model complexity, massive data, and large-scale deployments.
• Work closely with algorithm researchers to co-design infrastructure aligned with cutting-edge machine learning frameworks and hardware accelerators.
• Continuously improve model architecture, efficiency, and scalability to enable better user experiences.
• Abstract and build reusable infrastructure components and tools that benefit the broader recommendation stack.

Skills

GPU architecture knowledge

CUDA/CUTLASS/Triton Lang

TensorFlow or PyTorch

Independent & collaborative work

Education

Bachelor's degree or above in Computer Science or related field

Tools

MLIR

XLA

TVM

Job description

TikTok Inc. in San Jose is seeking talented Software Engineers to join our Model Infrastructure team. You will focus on architecture and engineering challenges for large-scale recommendation models, from online inference to offline training, to support evolving foundation models.

You will co-design infrastructure with researchers to balance compute cost with recommendation effectiveness, optimize performance, and enable scalable experimentation across GPU clusters.