Software Engineer in Machine Learning Infra - TikTok Recommendation Architecture

TikTok is seeking a Software Engineer in Machine Learning Infra to join their Recommendation Architecture Team in Singapore. The team is responsible for building and optimizing the architecture for the TikTok recommendation system, ensuring stability, high availability, and performance. TikTok’s mission is to inspire creativity and bring joy through its platform, fostering a culture of innovation and collaboration.

Requirements

• Bachelor's degree or above in Computer Science or related field
• 3+ years of experience building scalable systems
• Experience in at least one or two programming languages in a Linux environment (C/C++/golang)
• Understanding of GPU hardware architecture and software stack (CUDA, cuDNN)
• Experience in deep model inference/training, debugging, and tuning
• Familiarity with model optimization tools (TVM, MLIR, XLA)
• Familiarity with mainstream machine learning frameworks (Tensorflow, Pytorch, MxNet)