Research Engineer Graduate (AI Training Systems Reliability & Performance - Seed Infra) - 2026 Start (PhD)

About the Team
The Seed Infrastructures team oversees the distributed training, reinforcement learning framework, high-performance inference, and heterogeneous hardware compilation technologies for AI foundation models.

We are looking for talented individuals to join our team. As a graduate, you will get opportunities to pursue bold ideas, tackle complex challenges, and unlock limitless growth.
Successful candidates must be able to commit to an onboarding date by the end of the year. Please state your availability and graduation date clearly in your resume.

Responsibilities
- Improve the reliability and performance of large-scale training systems across pre-training, fine-tuning, evaluation, and inference
- Build observability, profiling, and debugging tools for distributed ML workloads
- Identify and optimize performance bottlenecks across GPU, networking, and storage layers
- Contribute to distributed training frameworks in multi-GPU and multi-node environments
- Collaborate with model and infrastructure teams to improve system scalability and efficiency
- Support incident analysis and operational stability

The base salary range for this position in the selected city is $241680 - $456000 annually.