ByteDance
Student Researcher - (Seed Model - AI Foundation Models Infrastructure) - 2026 Start (PhD)
Seattle, Washington · Internship · Posted Oct 28, 2026
- Education
- Currently pursuing a PhD degree in Computer Science, Electrical Engineering, or related technical fields.
About the job
Research and develop infrastructure for AI foundation models, including distributed training, reinforcement learning, inference, and heterogeneous hardware optimization.
What you'll do
- Design and optimize distributed training and post-training systems.
- Improve foundation-model inference performance and develop compiler or runtime optimizations for heterogeneous hardware.
- Analyze system performance and build tooling to improve developer productivity and reliability.
What they ask for
- Currently pursuing a PhD in Computer Science, Electrical Engineering, or a related technical field.
- Strong programming skills in Python and/or C++.
- Solid understanding of systems, distributed computing, machine learning systems, or performance optimization.
- Experience with at least one of distributed training frameworks, reinforcement learning training systems, GPU programming, or compiler technologies.
Nice to have
- Experience with large-scale ML systems or infrastructure projects; open-source ML systems or performance tooling contributions.
- Publications in ML systems, distributed systems, or related areas (a plus but not required).
Tools
Python, C++, PyTorch FSDP, Megatron-style parallelism, CUDA, Triton
AI focus
Develop infrastructure for AI foundation models, including distributed training, reinforcement learning training, inference, and hardware optimization.
Model-written summary · the original posting is authoritative