Student Researcher (AI Foundation Models Infrastructure - Seed Infra) - 2026 Start (PhD)
ByteDanceResponsibilities
- Design and optimize large-scale distributed training systems, including parallelism, memory efficiency, and fault tolerance.
- Contribute to reinforcement learning training frameworks and large-scale post-training systems.
- Improve foundation-model inference performance, latency, and throughput.
- Develop compiler or runtime optimizations for heterogeneous hardware such as GPUs and accelerators.
- Perform system-level performance analysis, profiling, and bottleneck diagnosis.
- Build tooling and automation to improve developer productivity and system reliability.
Requirements
- 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 distributed training frameworks, reinforcement learning training systems, GPU programming, or compiler technologies.
- Preferred: experience with large-scale ML systems or infrastructure projects, open-source ML systems or performance tooling contributions, and publications in ML systems or distributed systems.
Categories
About ByteDance
ByteDance is a global incubator of platforms at the cutting edge of commerce, content, entertainment and enterprise services - over 2.5bn people interact with ByteDance products including TikTok. Creation is the core of ByteDance's purpose. Our products are built to help imaginations thrive. This is doubly true of the teams that make our innovations possible. Together, we inspire creativity and enrich life - a mission we aim towards achieving every day. At ByteDance, we create together and grow together. That's how we drive impact - for ourselves, our company, and the users we serve. We are committed to building a safe, healthy and positive online environment for all our users. We have over 110,000 employees based in more than 30 countries globally. Join us.