Research Engineer Graduate (AI Training Systems & RL Infrastructure - Seed Infra) - 2026 Start (PhD)
ByteDanceResponsibilities
- Conduct research and development on large-scale AI infrastructure for training and post-training foundation models, multimodal LLMs, and image/video generation models.
- Design and optimize distributed training strategies, including data, model/tensor, pipeline, and expert parallelism, computation-communication overlap, and GPU cluster scaling.
- Prototype and improve end-to-end reinforcement learning training systems covering rollout generation, policy optimization, evaluation, and iterative deployment.
- Build scalable, fault-tolerant infrastructure for dynamic workloads and heterogeneous compute environments.
- Analyze networking, scheduling, and GPU memory bottlenecks and develop optimizations for throughput, efficiency, and stability.
- Develop tooling, monitoring, debugging, and observability frameworks for large-scale training and reinforcement learning systems.
- Collaborate on system-algorithm co-design and translate research prototypes into production-ready infrastructure.
Requirements
- Completing or recently completed a PhD in Computer Science, Electrical Engineering, Electrical and Computer Engineering, Physics, Mathematics, or a related discipline.
- Strong background in distributed systems, large-scale machine learning systems, or deep learning infrastructure.
- Research or hands-on experience training or optimizing large-scale models, including LLMs, multimodal models, or reinforcement learning systems.
- Understanding of data, model/tensor, pipeline, and expert parallelism and distributed training concepts.
- Familiarity with reinforcement learning workflows such as rollout generation, policy optimization, and evaluation loops.
- Proficiency in Python and/or C++ and experience with modern machine learning frameworks such as PyTorch and distributed training tools.
- Ability to state availability and graduation date clearly and commit to an onboarding date by the end of the year.
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.