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