AI Graphics Research Scientist & Software Engineer

TikTok; San Jose (CA); On-site; USD 162,000 - 388,000; Full time

Job summary

TikTok is seeking a PhD-level researcher to join the Engine & Tools AI-3D team in California. You will work on cutting-edge 3D generation, animation, and neural rendering, translating research into interactive engine pipelines.

We value strong ML foundations, publications, and hands-on experience with PyTorch or JAX, plus a passion for AI-for-graphics and scalable experimentation in a fast-paced environment.

Qualifications

• PhD in CS/CE/EE or related field completed or near completion.
• Strong research background with publications, internships, or significant projects.
• Hands-on experience training, fine-tuning, or evaluating ML models with PyTorch or JAX.
• Familiarity with modern generative architectures and training methods.
• Ability to work with software engineers to move research toward production.

Responsibilities

• Join TikTok Engine & Tools AI-3D team to advance AI-driven 3D graphics.
• Develop, train, and evaluate models for 3D generation, rendering, and animation.
• Contribute to data pipelines, toolchains, and experimental frameworks for scalable experiments.

Skills

PhD in CS/CE/EE/related

Research experience

ML foundations (PyTorch/JAX)

Generative modeling familiarity

Model development workflows

AI-for-graphics knowledge

Experiment design & evaluation

Education

PhD in computer science/related field

Tools

PyTorch

JAX

Job description

TikTok is seeking a PhD-level researcher to join the Engine & Tools AI-3D team in California. You will work on cutting-edge 3D generation, animation, and neural rendering, translating research into interactive engine pipelines.

We value strong ML foundations, publications, and hands-on experience with PyTorch or JAX, plus a passion for AI-for-graphics and scalable experimentation in a fast-paced environment.