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

Note: The job is a remote job and is open to candidates in USA. TikTok is the leading destination for short-form mobile video, and they are seeking a Machine Learning Engineer Intern to join their Graphics Engine Team. The role involves designing and implementing advanced algorithms for 3D generation, developing deep learning models, and optimizing systems for performance across various platforms.

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

Skills

• Currently pursuing an Undergraduate/Graduate in Software Development, Computer Science, Computer Engineering, or a related technical discipline
• Strong background in linear algebra, calculus, geometry, and numerical methods
• Solid experience in one or more of the following: neural rendering (NeRF, 3DGS, or similar), physically based simulation, animation system, AI
• Proficiency in C++ and Python
• Familiarity with GPU programming (CUDA, OpenCL, or similar) is a plus, but not required
• Experience with deep learning frameworks (e.g., PyTorch, TensorFlow) and applying neural networks to graphics, simulation, or generative models is a strong plus
• 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

• Interns have day one access to health insurance
• Life insurance
• Wellbeing benefits and more
• Interns also receive 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)
• Interns who are not working 100% remote may also be eligible for housing allowance.

Company Overview

Company H1B Sponsorship