Apply on Employer Site

TikTok · 2 weeks ago

Machine Learning Engineer Intern (TikTok-Engine and Tools-3D graphics) - 2026 Start (PhD)

San Jose, CA

Internship

Onsite

Intern

$60/hr - $60/hr

Start in 2026

TikTok is the leading destination for short-form mobile video, and they are seeking a Machine Learning Engineer Intern for a 2026 start. This role involves designing and implementing algorithms for 3D generation and editing, developing deep learning models, and optimizing systems for performance across various hardware.

Content CreatorsContent DiscoveryMedia and EntertainmentSocial MediaVideo

H1B Sponsor Likely

Responsibilities

Design and implement advanced algorithms for 3D generation/editing, real-time physics simulation, character animation, geometric modeling, rendering etc

Develop and integrate deep learning models that enhance speed, fidelity, or realism in physically based simulations or complex graphics workflows

Optimize algorithms and systems for speed and scalability across a wide range of hardware and platform targets

Work closely with product or other engineering teams to identify key challenges and deliver seamless, high-impact graphics solutions

Stay up to date with the latest research in computer graphics, physics simulation, and machine learning. Contribute to internal knowledge-sharing and may publish findings in leading conferences or journals

Qualification

Machine LearningComputer GraphicsProgramming SkillsMathematical FoundationsDeep Learning FrameworksGPU Programming3D ModelingAnimation SystemsAI

Required

Currently pursuing an Undergraduate/Graduate in Software Development, Computer Science, Computer Engineering, or a related technical discipline

Mathematical Foundations: Strong background in linear algebra, calculus, geometry, and numerical methods

Computer Graphics: Solid experience in one or more of the following: neural rendering (NeRF, 3DGS, or similar), physically based simulation, animation system, AI

Programming Skills: Proficiency in C++ and Python. Familiarity with GPU programming (CUDA, OpenCL, or similar) is a plus, but not required

Machine Learning: Experience with deep learning frameworks (e.g., PyTorch, TensorFlow) and applying neural networks to graphics, simulation, or generative models is a strong plus

Preferred

Graduating December 2026 onwards with the intent to return to degree program after the completion of the internship

Relevant publications in top-tier computer graphics, vision, or ML conferences/journals (e.g., SIGGRAPH, ICCV, CVPR, NeurIPS)

A solid background in both 3D computer graphics and AI is preferred

Benefits

Day one access to health insurance

Life insurance

Wellbeing benefits

10 paid holidays per year

Paid sick time (56 hours if hired in first half of year, 40 if hired in second half of year)

Housing allowance

Company

TikTok

Glassdoor3.4

TikTok is a short-form video entertainment app and social network platform. It is a sub-organization of ByteDance.

Founded in 2003

Los Angeles, California, USA

10001+ employees

https://www.tiktok.com

H1B Sponsorship

TikTok has a track record of offering H1B sponsorships. Please note that this does not guarantee sponsorship for this specific role. Below presents additional info for your reference. (Data Powered by US Department of Labor)

Distribution of Different Job Fields Receiving Sponsorship

Represents job field similar to this job

Trends of Total Sponsorships

2025 (979)

2024 (601)

2023 (387)

2022 (322)

2021 (133)

2020 (72)