Home

Tech Lead Machine Learning Engineer - AIGC Model, TikTok Monetization GenAI at TikTok

Location: San Jose, CA, US

Job Summary:

Job Duties and Scopes
- Develop and implement AI-driven applications for text-to-image and text-to-video content generation.
- Design and enhance multi-GPU systems for efficient training and deployment of generative models.
- Collaborate with cross-functional teams to optimize platform performance, focusing on latency reduction.
- Stay updated with advancements in generative AI to integrate new technologies.

Required Skills
- Deep expertise in Python and deep learning frameworks (e.g., PyTorch).
- Proficiency in Linux, Docker, and Kubernetes environments.
- Strong understanding of generative models, particularly text-to-image and text-to-video technologies.

Required Experiences
- M.S./Ph.D. in Computer Science or related fields.
- Proven experience with large-scale model training technologies (e.g., FSDP/DeepSpeed).
- Experience with cloud computing and distributed systems, preferably in GCP/Azure environments.

Job URLs: