Research Engineer, Generative AI Model Optimization

am is a team focusing on building advanced end-to-end production pipelines with core technologies including generative AI, intelligent editing and content understanding. By utilizing AI model training, optimization, deployment and applications, we provide cutting-edge AI capabilities to empower content creation and consumption on TikTok and serve billions of users.

We are looking for strong research engineers to design and develop state-of-the-art generative AI model optimization systems and solutions. The primary goal is to reduce the cost of generative AI applications while preserving quality and efficiency.

Responsibilities

• Design and develop state-of-the-art optimization framework and solutions for generative AI model training and inference, utilize techniques including but not limited to parallelism, quantization, distillation and so on.
• Research and develop cutting-edge distributed model training and inference optimization technologies, such as data parallelism, model parallelism etc.
• Communicate and collaborate with cross-functional teams to clarify model performance requirements under different use cases, make reasonable trade-offs to provide the most cost-efficient model optimization solutions to TikTok product teams to support content creation and consumption.
• Built platforms and tools that standardize common optimization methods to enlarge the usability of model optimization technologies among internal developers and researchers.

Qualifications

Minimum Qualifications

• M.S. or Ph.D. in Computer Science or related fields with 3-5 years of experience in software development.
• Proficient in one or more programming languages, such as Python, C, or C++. Familiar with common data structures and basic algorithms.
• Deep knowledge in machine learning optimizations such as quantization, pruning, knowledge distillation, Neural Architecture Search (NAS), and optimizations for large models.
• Familiar with one or more open-source deep learning frameworks, such as PyTorch, TensorFlow, PaddlePaddle etc. Understand the underlying design of the framework, familiar with common model architectures.

Preferred Qualifications

• Experience in AutoML algorithms or model acceleration technologies.
• Experience in contributing to open-source projects is preferred.
• Ability to work collaboratively with global stakeholders

TikTok is committed to creating an inclusive space where employees are valued for their skills, experiences, and unique perspectives. Our platform connects people from across the globe and so does our workplace. At TikTok, our mission is to inspire creativity and bring joy. To achieve that goal, we are committed to celebrating our diverse voices and to creating an environment that reflects the many communities we reach. We are passionate about this and hope you are too.

Job Information

[For Pay Transparency] Compensation Description (annually)

The base salary range for this position in the selected city is $136800 - $280000 annually.

Compensation may vary outside of this range depending on a number of factors, including a candidate's qualifications, skills, competencies and experience, and location. Base pay is one part of the Total Package that is provided to compensate and recognize employees for their work, and this role may be eligible for additional discretionary bonuses/incentives, and restricted stock units.

Our company benefits are designed to convey company culture and values, to create an efficient and inspiring work environment, and to support our employees to give their best in both work and life. We offer the following benefits to eligible employees:

We cover 100% premium coverage for employee medical insurance, approximately 75% premium coverage for dependents and offer a Health Savings Account(HSA) with a company match. As well as Dental, Vision, Short/Long term Disability, Basic Life, Voluntary Life and AD&D insurance plans. In addition to Flexible Spending Account(FSA) Options like Health Care, Limited Purpose and Dependent Care.

Our time off and leave plans are: 10 paid holidays per year plus 17 days of Paid Personal Time Off (PPTO) (prorated upon hire and increased by tenure) and 10 paid sick days per year as well as 12 weeks of paid Parental leave and 8 weeks of paid Supplemental Disability.

We also provide generous benefits like mental and emotional health benefits through our EAP and Lyra. A 401K company match, gym and cellphone service reimbursements. The Company reserves the right to modify or change these benefits programs at any time, with or without notice.