Note: The job is a remote job and is open to candidates in USA. Axelera AI is a deep-tech startup creating the next-generation AI platform to support advancements in technology. The Senior/Staff Engineer – Computing Architecture will design and develop AI computing systems, working closely with various teams to optimize performance and scalability for cutting-edge solutions.

Responsibilities

• Architecture Design: Define and develop computing architectures for AI accelerators
• Performance Optimization: Analyze and optimize computational workloads, ensuring low power consumption, high efficiency, and maximum performance for AI applications
• Technology Leadership: Drive innovation in architectural design by evaluating emerging technologies, industry trends, and academic advancements
• Cross-Functional Collaboration: Work with hardware, software, and product teams to translate system requirements into architectural solutions that balance performance, cost, and power
• System Modeling & Tooling Development: Develop simulation and modeling tools to validate architectural designs and predict system performance across target AI workloads
• Mentorship: Provide technical guidance and mentorship to junior engineers, fostering a culture of excellence and innovation within the team

Skills

• Master's or PhD in Computer Engineering, Electrical Engineering, or a related field
• 5+ years of experience in computing architecture or system design, preferably within AI, semiconductor, or computing domains
• Strong expertise in AI accelerator designs, including neural processing units (NPUs) or graphics processing units (GPUs)
• Proficiency in architectural modeling, simulation tools, and performance analysis frameworks
• Proficiency in Python or C++ programming with a strong understanding of language features and best practices
• Deep understanding of state-of-the-art AI inference workloads, algorithms, and dataflow
• Proven track record of delivering complex computing architectures from concept to implementation
• Strong communication and collaboration skills, with the ability to work in a multidisciplinary and global environment
• Familiarity with the modern IC design flow from architecture to GDSII to communicate with silicon implementation stakeholders
• Familiarity with parallel programming models, e.g. via CUDA or similar APIs
• Hands-on experience with low-level programming close to hardware and understanding of computer system architecture basics
• Experience with AI compilation frameworks such as MLIR and/or TVM

Benefits

• Pension plan
• Extensive employee insurances
• Option to get company shares

Company Overview