Graduate Research Scientist, Neural Graphics & World Models

TikTok; San Jose (CA); On-site; USD 128,000 - 317,000; Full time

Benefits offered by this job

Medical, dental, vision insurance

401(k) with company match

Paid time off

Job summary

TikTok is seeking new-graduate Research Scientists/Engineers to work on neural graphics and world models. You will own a scoped project within a research focus track and collaborate with engineers to connect models to data, runtime, and engine pipelines.

The role emphasizes turning ideas into working experiments, analyzing failures, and building prototypes, with a path toward interactive engine workflows and graphics tooling.

Qualifications

• PhD completed or near-complete in CS/CE/related field.
• Experience via lab, thesis, publication, internship, open-source project, or substantial independent project.
• Strong foundation in machine learning, generative modeling, and experimental methodology; hands-on with training, fine-tuning, or evaluating models using PyTorch or JAX.
• Familiarity with modern generative architectures and training methods (VAEs, diffusion, autoregressive, multimodal).
• Engineering fluency with model development workflows and evaluating results.
• Deep experience in AI-for-graphics areas: 3D modeling, asset generation, rendering, world models, or graphics/game-engine workflows.

Responsibilities

• Study relevant research, reproduce strong baselines, and identify practical extensions.
• Develop and evaluate models, data methods, or evaluation methods in a scoped research area.
• Build components of training, inference, data, or evaluation workflows using modern ML frameworks.
• Run controlled experiments, analyze model behavior, and communicate findings clearly.
• Create research prototypes for interactive graphics, creative tools, or engine-based workflows.
• Produce reproducible research artifacts, clear documentation, demonstrations, and technical reports.

Skills

Machine learning

Generative modeling

Experimental methodology

PyTorch

JAX

Education

PhD in computer science or related field

Tools

PyTorch

JAX

Job description

TikTok is seeking new-graduate Research Scientists/Engineers to work on neural graphics and world models. You will own a scoped project within a research focus track and collaborate with engineers to connect models to data, runtime, and engine pipelines.

The role emphasizes turning ideas into working experiments, analyzing failures, and building prototypes, with a path toward interactive engine workflows and graphics tooling.