ByteDance
Student Researcher - (Seed Model - AI Foundation Model Infrastructure) - 2026 Start (PhD)
San Jose, California · 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, heterogeneous hardware compilation, and system performance.
What you'll do
- Design and optimize distributed training systems and post-training frameworks.
- Improve foundation model inference performance and develop compiler or runtime optimizations for heterogeneous hardware.
- Analyze system performance and build tools 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 area: 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
AI foundation model infrastructure, including distributed training, reinforcement learning training, inference optimization, and heterogeneous hardware compilation.
Model-written summary · the original posting is authoritative