AI Computing Architecture Researcher

In this role, you will focus on designing and optimizing next-generation AI computing architectures and platforms for LLM, AIGC, distributed AI workloads, and intelligent agent systems. You will collaborate with researchers and engineers to improve the performance, scalability, and efficiency of AI infrastructure across GPU, NPU, and heterogeneous computing environments. 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 Qualifications: Master’s or PhD in Computer Science, Electrical Engineering, or related fields, 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. Experience with AI/deep learning frameworks and tools for performance profiling and optimization, including optimization in agent systems. Knowledge of low-level hardware optimization, including memory management and instruction-level tuning for CPU/GPU/NPU architectures. Familiarity with AI compiler and AI computing architecture, such as LLVM, MLIR, TVM. Background in designing and implementing high-performance distributed systems and storage solutions, including those supporting agent coordination and skill sharing. Strong understanding of networking, I/O, and data management techniques for AI and agent workloads. Exposure to multi-agent system design, including communication protocols, coordination mechanisms, skill composition frameworks techniques, harness engineering for AI workloads. Interested candidate please click "APPLY" to begin your job search journey. We regret to inform that only shortlisted candidates will be notified. By sending us your personal data and curriculum vitae (CV), you are deemed to consent to PERSOL Singapore Pte Ltd and its affiliates to collect, use and disclose your personal data for the purposes set out in the Privacy Policy available at https://www.persolsingapore.com/policies. You acknowledge that you have read, understood, and agree with the Privacy Policy. PERSOL Singapore Pte Ltd •RCB No. 200007268E • EA License No. 01C4394 •Registration ID: Heah Sian Wei R23117518