About the Role
ByteDance’s Seed Infrastructures team focuses on building cutting-edge infrastructure for AI foundation models, including distributed training and high-performance inference. This role offers an opportunity to contribute to advancing artificial general intelligence, powering diverse applications, and driving technological progress for society.
Responsibilities
- Optimize large-scale distributed training systems for efficiency and reliability.
- Develop and improve reinforcement learning training pipelines and post-training systems.
- Enhance inference performance, including latency, throughput, and stability.
- Implement compiler or runtime optimizations for GPU and other accelerators.
Requirements
- Currently pursuing a Bachelor’s or Master’s degree in Computer Science, Electrical Engineering, or related technical fields.
- Familiarity with machine learning frameworks such as PyTorch or similar tools.
- Proficiency in Python or C++.
- Strong analytical and problem-solving abilities.