Staff AI Compiler Engineer
5 days ago
Description
• Lead the design, architecture, and hands-on development of Axelera AI’s Neural Network compiler and SDK. • Drive innovation by designing compiler optimizations, focusing on hardware-aware transformations for In-Memory Computing. • Execute and technically guide complex projects to ensure our compiler supports workloads beyond classical CNNs (e.g. generative AI) and upcoming generations of our heterogeneous compute platform. • Collaborate with cross-functional teams to ensure alignment on technical strategy, optimizing the Axelera software stack for scalability and performance. • Oversee holistically our compilation pipeline and proactively propose changes to ensure best-in-class functional coverage, performance and accuracy. • Ensure best practices in the team, especially regarding testing methodologies for compilers. • Mentor and guide junior and senior engineers, modeling a standard of excellence and leadership within the team, thereby being a valuable partner to your manager and peers.
Requirements
• Master’s or PhD in Computer Science, Electrical Engineering, or a related technical field. • 5+ years of experience in software engineering, with at least 2 years focused on Deep Learning and AI systems. • Strong experience in AI compiler development, e.g. in some of these topics: Workload partitioning for heterogeneous platforms, scheduling and tiling optimizations, managing a multi-level memory hierarchy, synchronization. • Background in computer architecture with related knowledge of heterogeneous compute systems. • Familiarity with open-source compiler frameworks such as TVM, MLIR, or Glow. • Experience in neural network quantization techniques is a plus. • Proficiency in PyTorch, with a good understanding of deep learning frameworks is a plus. • Excellent communication and leadership skills, with a high sense of ownership and the ability to drive technical solutions that impact the broader organization. • Proven to be a team player favoring collaborative decision making over building silos. • Proficiency in C++ and Python.
Benefits
• Attractive compensation package • Pension plan • Extensive employee insurances • Option to receive company shares
Apply Now