Student Researcher (AI Foundation Models Infrastructure – Seed Infra) – 2026 Start

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.

Responsibilities
- As an Infrastructure Intern, you may work on one or more of the following areas:
- Assist in building and optimizing large-scale distributed training systems (e.g., data/model parallelism, memory efficiency, reliability)
- Support the development and improvement of reinforcement learning training pipelines and post-training systems
- Improve inference performance, including latency, throughput, and system stability
- Contribute to compiler or runtime optimizations for GPU and other accelerators
- Conduct performance analysis, profiling, benchmarking, and bottleneck identification
- Develop internal tools and automation to improve infrastructure efficiency and developer productivity
- Collaborate with researchers and engineers to translate model requirements into scalable system solutions

annually.