Internship AI Software stack & Compilation

About Us: At Huawei Technologies Switzerland AG we are a leading technology firm dedicated to developing cutting-edge solutions that redefine industry standards and push technological boundaries. Our core focus is on creating advanced computing architectures that can efficiently support and enhance the performance of artificial intelligence systems. We believe in innovation as a driving force for improvement and are committed to achieving excellence in all areas of research and development.

Job Summary: We are seeking a highly skilled and motivated Senior Researcher to join our team focusing on the development and optimization of AI software stacks and compilation techniques. The successful candidate will have substantial experience with state-of-the-art frameworks and be instrumental in advancing our research and product offerings in AI model performance and efficiency.

Key Responsibilities:

Conduct cutting-edge research in AI software stacks focusing on performance optimization and compiler technologies.

• Develop and enhance tools and frameworks specifically utilizing PyTorch Triton MLIR etc.

• Collaborate with cross-functional teams to integrate research findings into practical applications and products.

• Publish research findings in top-tier conferences and journals to contribute to the academic community.

• Mentor junior researchers and engineers fostering an environment of collaboration and innovation.

• Stay updated on industry trends and advancements in AI and compiler technologies.

Qualifications:

Master or PhD students in Computer Science Electrical Engineering or a related field with a focus on AI compilers or software systems.

• Proven experience and proficiency in Python PyTorch Triton MLIR and TVM.

• Strong understanding of programming model compiler design program optimization and parallel processing.

• Experience with AI model training and deployment at scale.

• Excellent problem-solving skills and ability to work in a fast-paced collaborative environment.

• Strong communication skills with a track record of publishing research and presenting findings to diverse audiences.

Preferred Qualifications:

Experience in performance benchmarking and profiling of AI models.

• Familiarity with hardware architectures and their impact on AI model performance.

• Contributions to open-source projects or active involvement in the AI research community.
• Can work full time for at least 6 months
• Chinese-speaking students are preferred

What We Offer:

• Competitive salary and benefits package.

• Opportunities for professional growth and development.

• Be part of innovative projects that make a difference.

Access to state-of-the-art technology and tools.