Senior LLM Researcher & Architect for Scalable AI

TikTok; San Jose (CA); On-site; USD 308,000 - 588,000; Full time

Benefits offered by this job

Medical, dental, and vision insurance

401(k) with company match

Paid parental leave

Disability coverage

Job summary

TikTok is seeking researchers in LLM, VLM and Omni Model domains to advance pretraining, evaluation, data processing and RL-based alignment for scalable industry applications. You will lead incubation of high-capacity LLM solutions and collaborate with cross-functional teams.

You will design data recipes, explore new architectures, and translate research into product impact, working across content moderation, search and recommendations domains.

Qualifications

• Ph.D in Computer Science, Data Science, Artificial Intelligence, or a related field.
• Proficiency in programming languages such as Python, Rust, or C++ and a track record of working with deep learning frameworks (e.g., pytorch, deepspeed, megatron, vllm, etc.).

Responsibilities

• Lead the incubation of next-generation, high-capacity LLM solutions for TikTok business, identify and define both short and medium term objectives;
• Design methods, tools, data recipes and experiments to push forward state-of-art in large language models;
• Explore new model architecture and inference-efficient model design for LLM applications to scale impact on business
• Work closely with cross-functional teams to plan and implement projects harnessing LLMs for diverse purposes and vertical domains
• Extend the insights and impact from industry to academia

Skills

Python

Rust

C++

Education

PhD in Computer Science / AI or related field

Tools

PyTorch

DeepSpeed

Megatron

VLLM

Job description

TikTok is seeking researchers in LLM, VLM and Omni Model domains to advance pretraining, evaluation, data processing and RL-based alignment for scalable industry applications. You will lead incubation of high-capacity LLM solutions and collaborate with cross-functional teams.

You will design data recipes, explore new architectures, and translate research into product impact, working across content moderation, search and recommendations domains.