LLM Post-Training Engineer (Graduate) - Research to Product

TikTok; San Jose, Northern (CA, KY); Hybrid; USD 128,000 - 317,000; Full time

Benefits offered by this job

Medical insurance

Dental insurance

Vision insurance

401(k) with company match

Paid parental leave

Disability insurance

Life insurance

Wellbeing benefits

10 paid holidays

10 paid sick days

Paid Personal Time

Job summary

TikTok in San Jose is seeking graduates to join the Multimedia AI team, working on post-training strategies and model alignment for our global products, including video understanding, translation, and content classification. You will contribute to building robust evaluation pipelines and experiments.

The role blends research and production, requiring strong Python skills, experience with PyTorch or JAX, and familiarity with transformer models.

Qualifications

• Bachelor's degree in Computer Science, Machine Learning or a related discipline.
• Strong programming skills in Python and experience with deep learning frameworks such as PyTorch or JAX.
• Foundational understanding of Transformer architectures and LLM training principles.
• Demonstrated ability to learn quickly in a fast-paced environment with passion for AI application.

Responsibilities

• Support the development and optimization of post-training strategies, including instruction tuning, preference tuning (SFT/DPO/PPO), and model alignment.
• Assist in building robust evaluation pipelines to measure model performance, helpfulness, and safety across diverse multimedia product use cases.
• Participate in the research and implementation of cutting-edge methodologies in reward modeling and human preference learning.
• Collaborate with engineering teams to bridge the gap between experimental research and production-ready AI applications (e.g., video understanding, translation, and content classification).
• Analyze and process large-scale datasets to identify patterns that improve model behavior and alignment quality.

Skills

Python

PyTorch

JAX

Transformer

LLM training

Education

Bachelor's degree in Computer Science, Machine Learning or related discipline

Tools

DeepSpeed

Megatron-LM

Job description

TikTok in San Jose is seeking graduates to join the Multimedia AI team, working on post-training strategies and model alignment for our global products, including video understanding, translation, and content classification. You will contribute to building robust evaluation pipelines and experiments.

The role blends research and production, requiring strong Python skills, experience with PyTorch or JAX, and familiarity with transformer models.