AI Computing Architecture Researcher

Beijing Foreign Enterprise Management Consultants Co.,Ltd.; Singapore; On-site; SGD 120,000 - 180,000; Full time

Job summary

Huawei is seeking an AI Computing Architecture Researcher to design and optimize AI computing architectures and platforms for LLM, AIGC, and distributed AI workloads.

Responsibilities include interfacing software with GPUs/NPUs, improving performance, and collaborating with AI researchers to enhance platform efficiency and scalability across architecture and deployment layers.

Qualifications

• Master’s or PhD in Computer Science, Electrical Engineering, or related fields (PhD preferred).
• Strong experience in high-performance AI systems or computing platforms.
• Proficiency in C/C++, Python, AI-DSL with focus on low-level programming.
• In-depth knowledge of model optimization techniques (quantization, pruning, sparsity).
• In-depth knowledge of parallel programming and distributed systems.
• Experience with GPU/NPU programming (CUDA, Triton, cuTile).
• Familiarity with ML frameworks (PyTorch, TensorFlow, MXNet).
• Experience with containerization and cloud deployment (Docker, Kubernetes, SGLang, vLLM, HuggingFace).

Responsibilities

• Design and optimize AI computing architectures and platforms for LLM, AIGC, distributed AI, and intelligent agent workloads.
• Develop software components interfacing with GPUs, NPUs, and AI chips.
• Improve performance and efficiency of distributed computing and acceleration.
• Collaborate with AI researchers to enhance platform performance for complex AI workloads.
• Design and implement tools/frameworks for deployment, monitoring, scaling, and orchestration of AI workloads.
• Integrate new AI models and agent controllers into the computing architecture for scalability and fault tolerance.
• Conduct profiling, benchmarking, and optimization to achieve high throughput and low latency.
• Stay updated on AI hardware and agent technologies and improve platform capabilities.

Skills

C/C++

Python

AI-DSL

Low-level programming

Parallel programming

Distributed systems

Multi-agent coordination

GPU/NPU programming

PyTorch

TensorFlow

MXNet

CUDA

Triton

Docker

Kubernetes

SGLang

vLLM

HuggingFace

AWS

GCP

Azure

Education

Master’s or PhD in CS/EE (PhD preferred)

Tools

CUDA

Triton

cuTile

Docker

Kubernetes

SGLang

vLLM

HuggingFace

AWS

GCP

Azure

Job description

On behalf of Huawei, a world-renowned information and communication technology company, we are seeking passionate and talented individuals to join our team as AI Computing Architecture Researcher.

Key Responsibilities:

• Design and optimize AI computing architectures and platforms for LLM, AIGC, distributed AI, and intelligent agent workloads. Develop software components that interface with hardware accelerators such as GPUs, NPUs, and specialized AI chips.
• Improve performance and efficiency of distributed computing, parallel processing, and GPU/NPU acceleration.
• Collaborate with AI researchers to enhance platform performance for complex AI and agent-based applications and also to resolve system-level performance issues across architecture and platform components.
• Design and implement tools and frameworks for deployment, monitoring, scaling, and orchestration of AI workloads and agent systems.
• Integrate new AI models, agent controllers, and algorithms into the computing architecture and platform while ensuring scalability, fault tolerance, and efficiency.
• Conduct profiling, benchmarking, and performance optimization to achieve high throughput and low latency.
• Contribute to research on agent capabilities, including skill abstraction, composition, learning, and deployment.
• Stay updated on advancements in AI hardware, system architecture, and agent technologies, and continuously improve platform capabilities.

Required Skills and Qualifications:

• Master’s or PhD in Computer Science, Electrical Engineering, or related fields (PhD preferred), with strong experience in high-performance AI systems or computing platforms.
• Proficiency in C/C++, Python, AI-DSL with a focus on low-level programming for high-performance systems.
• In-depth knowledge of AI model optimization techniques such as quantization, model graph pruning, and model parameter compression and sparsity algorithm.
• In-depth knowledge of parallel programming, distributed systems, and multi-agent coordination strategies.
• Experience with GPU/NPU programming (e.g., CUDA, Triton, cuTile or similar DSLs).
• Familiarity with AI/machine learning frameworks such as PyTorch, TensorFlow, or MXNet.
• Understanding of AI model deployment, orchestration, and optimization on large-scale platforms, including agent skill deployment and runtime management.
• Experience with containerization (Docker, Kubernetes),LLM deployment platforms (SGLang, vLLM, HuggingFace), and cloud infrastructure (AWS, GCP, Azure).
• Strong problem-solving skills and the ability to optimize software performance for both traditional AI and agent-based workloads.