AI Infrastructure Intern: Distributed Training & ML Systems

Bytedance; San Jose, Northern (CA, KY); Hybrid; USD 20,000 - 31,000; Part time

Job summary

ByteDance Seed's Infrastructure team in San Jose is seeking an Infrastructure Intern within the Technology group. You will contribute to distributed training, RL pipelines, and high-performance AI infrastructure, gaining practical, on-the-job experience in a fast-paced environment.

You will collaborate with researchers and engineers to translate model requirements into scalable system solutions while developing core skills in ML systems during this short-term internship.

Qualifications

• Pursuing a Bachelor’s or Master’s degree in Computer Science, Electrical Engineering, or related technical fields.
• Proficiency in at least one programming language such as Python or C++.
• Familiarity with machine learning frameworks such as PyTorch.
• Strong analytical and problem-solving skills.
• Ability to work collaboratively in a fast-paced technical environment.
• Interest in pursuing long-term work in ML systems or AI infrastructure.

Responsibilities

• Assist in building and optimizing large-scale distributed training systems (e.g., data/model parallelism, memory efficiency, reliability).
• Support the development and improvement of reinforcement learning training pipelines and post-training systems.
• Improve inference performance, including latency, throughput, and system stability.
• Contribute to compiler or runtime optimizations for GPU and other accelerators.
• Conduct performance analysis, profiling, benchmarking, and bottleneck identification.
• Develop internal tools and automation to improve infrastructure efficiency and developer productivity.
• Collaborate with researchers and engineers to translate model requirements into scalable system solutions.

Skills

Python

C++

PyTorch

Analytical skills

Collaboration

ML systems interest

Education

Bachelor's or Master’s in CS/EE or related

Job description

ByteDance Seed's Infrastructure team in San Jose is seeking an Infrastructure Intern within the Technology group. You will contribute to distributed training, RL pipelines, and high-performance AI infrastructure, gaining practical, on-the-job experience in a fast-paced environment.

You will collaborate with researchers and engineers to translate model requirements into scalable system solutions while developing core skills in ML systems during this short-term internship.