Recommendation Architecture AI/ML I...

Role Overview

We are seeking a Graduate AI/ML Infrastructure Engineer to build and optimize distributed infrastructure for large-scale recommendation models and LLMs. This role involves training, online inference, GPU computing, and model serving within TikTok's San Jose, CA office.

Key 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 protocols.
• Collaborate with researchers to develop and deploy LLM training and serving solutions.
• Analyze system bottlenecks and implement strategic performance optimizations.