Axelera AI

Axelera AI develops purpose-built AI hardware acceleration technology for computer vision and generative AI applications.

51-200 employees
Founded 2021
$66.5M grant
Semiconductor Manufacturing
Private Company

Open Positions15

Europe (incl. UK)Full-TimeAI PlatformPosted
  • Design, implement, and maintain frontend and graph-level compiler components using MLIR
  • Develop and optimize graph-level transformations such as operator fusion, constant folding, operator sinking, graph partitioning, and other performance-critical optimizations
  • Extend and maintain MLIR dialects, passes, and infrastructure to support AI workloads
  • Integrate and lower AI models from frameworks such as PyTorch, ONNX, and TensorFlow into internal compiler representations
  • Collaborate with hardware and backend compiler teams to ensure efficient mapping of AI workloads to heterogeneous architectures
  • Support and mentor team members in adopting and effectively using MLIR infrastructure
  • Analyze model graphs and implement optimizations to improve performance, memory efficiency, and execution efficiency
  • Contribute to the design and evolution of the overall compiler architecture and tooling
  • Debug, profile, and improve compiler performance and correctness
PythonPyTorchC+++1 more
Showing 1 of 15 positions

About Axelera AI

Axelera AI is revolutionizing the landscape of artificial intelligence by bringing powerful, energy-efficient AI capabilities directly to the edge. This pioneering company develops a unique hardware and software platform designed to accelerate AI inference for critical real-world applications. Their Metis AI Platform, featuring the Metis AI Processing Unit (AIPU) and Voyager SDK, delivers a significant 3-5x increase in efficiency and performance compared to conventional cloud-based solutions. By enabling AI to operate on devices where data is generated, Axelera AI addresses key challenges of latency, privacy, and the immense energy consumption of centralized data centers. At Axelera AI, you will join a diverse and growing team of over 220 experts, including many PhDs, spread across 18 countries. The company fosters an open and collaborative culture where creativity and innovation thrive. They champion flexible working arrangements, offering both office-based roles in major European tech hubs and fully remote opportunities across the continent. This global, inclusive environment encourages collaborative ownership and empowers individuals to take responsibility, shaping the future of AI together. Technically, Axelera AI is at the forefront of hardware-software co-design. Their Digital In-Memory Computing (D-IMC) architecture is a core differentiator, integrating computing power directly within memory to minimize data transfer bottlenecks. This innovative approach allows their Metis AIPU to achieve impressive performance benchmarks with remarkable power efficiency, making advanced AI practical for diverse edge applications like industrial automation, smart cities, and robotics. With over $450 million in funding and a strong business pipeline, Axelera AI is rapidly scaling and expanding its product lines to meet the evolving demands of both edge and data center AI. If you are an impatient optimist and a maker at heart, eager to contribute to meaningful technological advancements, Axelera AI offers a unique opportunity. You will contribute to democratizing AI, putting powerful tools in the hands of millions of developers to reinvent the world around us. This is a chance to work with cutting-edge technology and a dedicated team, driving innovation and making a tangible impact in the burgeoning field of artificial intelligence.

Tech Stack

frameworkserverwidgetsweb_server