Machine Learning Engineer, Ads Creative - TikTok

Key Responsibilities

• Design and implement end‑to‑end ML pipelines for understanding and generating ad creative components such as video, music, and call‑to‑action elements.
• Apply computer‑vision and deep‑learning techniques to extract visual and audio features from large‑scale short‑form video datasets.
• Collaborate with product and engineering teams to integrate models into real‑time ad serving systems, ensuring low latency and high scalability.
• Conduct experiments, A/B tests, and performance monitoring to continuously improve model accuracy and business impact.
• Maintain and optimize cloud‑based training and inference infrastructure on AWS, including GPU provisioning and data pipelines.

Requirements

• Strong programming skills in Python and experience with TensorFlow or PyTorch.
• Hands‑on experience building computer‑vision or multimodal deep‑learning models for video and audio data.
• Proficiency in developing scalable ML pipelines and deploying models in production environments (AWS, Docker, Kubernetes).
• Solid understanding of statistical modeling, data preprocessing, and model evaluation techniques.
• Excellent problem‑solving abilities and the ability to work cross‑functionally in a fast‑paced environment.