Machine Learning Engineer Graduate (On-Device AI) - 2025 Start (PhD)

TikTok is the premier destination for short-form mobile videos, with a mission to inspire creativity and bring joy. The On-Device AI Team focuses on embedding advanced AI technologies into devices worldwide, enhancing features like visual enhancements and video editing tools. The company is looking for a talented Machine Learning Engineer Graduate to join their team in 2025.

In this role you can expect to have the responsibilities:

1. Develop and enhance the core On-Device AI engine for diverse platforms.
2. Utilize efficient machine learning techniques and hardware optimization to improve algorithmic performance.
3. Integrate state-of-the-art AI algorithms into TikTok and other products.
4. Translate emerging research into practical, user-centric applications.

This role requires you to have:

1. PhD in Computer Science or related field
2. 3+ years of experience in Python/C++
3. Solid CS fundamentals and problem-solving skills
4. 1+ years of experience in AutoML/model optimization
5. Excellent communication and teamwork skills

TikTok promotes an inclusive work environment that celebrates diverse voices. They provide reasonable accommodations for candidates with disabilities, pregnancy, and other protected reasons.