Compute System Arch AI Infra Intern - 2027

We are looking for a Compute System Arch AI Infra Intern to join our team. The Compute System Architect team's work scope covers the whole compute pipeline, memory system, and multi-GPU, CPU, and CPU interconnection. As an intern, you will have the opportunity to deeply learn the latest cross-unit new features in the new GPU architectures.

Responsibilities:

• Get familiar with Compute System Architect's daily work as background knowledge
• Get familiar with the team's existing AI infrastructure and flows
• Get a clear understanding of the AI infra requirement
• Co-work with mentor to propose, review, and finalize the design for the AI infra
• Efficiently implement the AI infra according to the design
• Actively collect testing use cases for the AI infra and use them to verify the implementation
• Deliver the AI infra to the team with well-organized documentation
• Collect and document feedback from users
• Proactively discuss improvement opportunities with mentor
• Provide a summary report out for the project

Requirements:

• Good at communication and collaboration
• Demonstrates strong analytical skills and a proven capacity for effective problem-solving
• Familiar with AI-assisted development
• Experience of building AI-integrated flows
• Pursuing a Bachelor in CS or EE. MS, PhD is a plus.

Nice to Have:

• Experience of building RAG and AI agent can be useful
• Knowledge of GPU architecture and/or experience of full-chip verification is helpful