Software Engineer Intern (AI Model Optimization) - 2026 Summer (BS/MS)

TikTok
Seattle, Washington

About the job

The Speech team's mission is to empower content interaction and creation using speech & audio related technologies. The team focuses on cutting-edge R&D in areas like speech & audio, music processing, and multimodal deep learning. We are looking for top talents to work on these exciting technologies, integrate them into various products and ultimately bring joy to our global user base! You will work on improving the performance and efficiency of large-scale AI models across training, inference, and deployment. This is an opportunity to gain hands-on experience in high-performance ML systems and contribute to enhancing user experience of all ByteDance-related products.

Responsibilities

Support research and engineering efforts to optimize deep learning models for speed, memory, and scalability.

Contribute to benchmarking and profiling tools to identify performance bottlenecks.

Collaborate with engineers to integrate optimized models into production pipelines.

Qualifications

Minimum

Currently pursuing a Bachelor's or Master's degree in Computer Science or a related field.

Must be able to commit to a 12-week full-time work period during Fall 2026

Currently pursuing a BS, or MS in Computer Science, Electrical Engineering, or related field.

Strong programming skills in Python and familiarity with deep learning frameworks (PyTorch, TensorFlow, JAX).

Knowledge of fundamental ML concepts and algorithms.

Must obtain work authorization in country of employment at the time of hire, and maintain ongoing work authorization during employment.

Preferred

Experience with AI model training/inference/deployment is a plus.

Familiarity with GPU programming (CUDA, Triton, or similar) is a plus.

Strong problem-solving skills and eagerness to learn.