Research Intern: Next-generation NPU and Agentic CPU Micro-architecture

Huawei Switzerland; Zürich; Vor Ort; CHF 120.000 - 170.000; Vollzeit

Zusammenfassung

Huawei Technologies Switzerland AG is seeking a highly skilled researcher to advance Neural Processing Unit architectures and agentic CPU designs. You will prototype, evaluate, and optimize frontend/back-end structures, memory hierarchies, and context management for AI workloads.

You will publish at premier venues and collaborate with leading universities, mentoring junior researchers. The role emphasizes strong development and prototyping capabilities, rigorous analysis, and the ability to work

Qualifikationen

• MSc in Computer Science, Electrical Engineering, or a related field.
• Strong background in Computer Architecture and Micro-architecture.
• Creativity and the ability to think outside the box to develop innovative technologies.
• Research experience in Computer Architecture/Micro-architecture, Vector/Matrix Extensions, GPU programming or profiling.
• Excellent analytical, problem-solving, and system-level thinking skills.
• Strong development and prototyping skills.
• Strong interpersonal skills and ability to work independently.

Aufgaben

• Research the next generation of Neural Processing Units (NPUs) core micro-architecture and CPU designs for agentic AI.
• Prototype new architectural features for NPUs, including frontend, backend, and memory systems.
• Explore NPU core micro-architecture details (branch prediction, BTB, prefetchers, register files, functional units).
• Investigate NPU backend and memory systems with TLBs, L1 caches, and scratchpads.
• Design CPUs for agentic AI including tool calling and context management.
• Develop resilience for bursty compute and rapid context switching beyond 2-way SMT.
• Publish research papers at top-tier venues (ASPLOS, ISCA, MICRO, HPCA).
• Collaborate with academic institutions and mentor junior researchers and interns.

Kenntnisse

Computer Architecture

Micro-architecture

GPU programming

PMU profiling

Analytical thinking

Prototyping skills

Independently collaborative

Ausbildung

MSc in Computer Science or Electrical Engineering

Tools

CUDA

Jobbeschreibung

Huawei envisions a world where technology connects people, empowers industries, and unlocks human potential. Guided by its mission to enrich lives through communication and intelligent innovation, Huawei stands at the forefront of global digital transformation. As a leader in Information and Communications Technology (ICT), the company pioneers breakthroughs in artificial intelligence, cloud computing, and smart devices—building the intelligent foundation of a fully connected world.

Through its Carrier, Enterprise, and Consumer business groups, Huawei delivers resilient digital infrastructure, advanced cloud and AI platforms, and transformative devices that enable progress at every level. Supporting 45 of the world’s top 50 telecom operators and serving one-third of the global population across more than 170 countries, Huawei is shaping a future where connectivity becomes a powerful catalyst for opportunity and sustainable growth.

This spirit of bold innovation is embodied by Huawei Technologies Switzerland AG. From its research hubs in Zurich and Lausanne, pioneering teams push the boundaries of High-Performance Computing, Computer Architecture, Computer Vision, Robotics, Artificial Intelligence, Neuromorphic Computing, Wireless Technologies, and Networking—architecting the intelligent systems that will define tomorrow’s digital era.

Responsibilities:

• The research investigates the next generation of Neural Processing Units (NPUs), with particular focus on core micro-architecture. This spans the frontend (branch prediction, BTB, instruction prefetchers), register files, issue and wake-up logic, scalar functional units, vector functional units, and tensor units; the backend (TLBs, L1, scratchpads); and the broader cache hierarchies and memory systems that feed them. In parallel, the work explores a new generation of CPUs tailored to the agentic AI era. As agent-based systems now spend time not only on the GPU/NPU but also on the CPU—handling tool calling, agent-logic scheduling, context management, and similar tasks—the CPU must become resilient to bursty compute and capable of rapid context switching, without the resource thrashing and limited on-chip contexts that constrain today's designs (most of which support only 2-way SMT).

• Investigate and prototype new architectural features, including but not limited to:

  • NPU Core Micro-architecture: Explore frontend mechanisms (branch prediction, BTB, instruction prefetching), register file organization, issue and wake-up logic, and the design of scalar, vector, and tensor functional units to maximize throughput and utilization for AI workloads.

  • NPU Backend and Memory System: Investigate backend structures including TLBs, L1 caches, and scratchpads, along with cache hierarchies and memory systems that sustain high bandwidth and low latency to the compute units.

  • Agentic CPU Architecture: Design CPUs tailored to agentic AI, where the processor handles tool calling, agent-logic scheduling, and context management alongside accelerator-bound work.

  • Resilience to Bursty Compute and Context Switching: Develop architectural support for rapid context switching and high thread-level concurrency that goes beyond conventional 2-way SMT, mitigating the on-chip resource thrashing and limited context capacity of current designs.

• Produce and present research papers at top-tier conferences and journals (e.g., ASPLOS, ISCA, MICRO, HPCA)

• Establish and maintain collaborations with leading academic institutions and faculty.

• Mentor and support junior researchers and interns in their professional development.

Requirements:

• MSc in Computer Science, Electrical Engineering, or a related field.

• Strong background in Computer Architecture and Micro-architecture is a must.

• Creativity and the ability to think outside the box to develop innovative technologies.

• Research experience in at least one of the following areas:

  • Computer Architecture and Micro-architecture: Computer Architecture and Micro-architecture of modern Intel/ARM/AMD CPUs, Nvidia GPUs, or Google’s TPUs.

  • Vector and Matrix Extensions: ARM SVE/SME or Intel SSE/AVX/AMX.

  • GPU programming model: Understanding of CUDA kernels and PTX/SASS instructions.

  • Micro-architecture characterisation: Understanding of profiling, characterisation, and bottleneck analysis of applications on CPU/GPUs via Performance Monitoring Units (PMUs).

• Excellent analytical, problem-solving, and system-level thinking skills.

• Strong development and prototyping skills are a must.

• Strong interpersonal skills, with a collaborative spirit and the ability to work independently.

Why join us:

• Collaborate with world-class scientists and engineers in an open, curiosity-driven environment;

• Access to state-of-the-art technology and tools;

• Opportunities for professional growth and development;

• Competitive salary and a high quality of life in Zurich, at the centre of Europe;

• Last but certainly not least: be part of innovative projects that make a difference.