The Senior/Staff Applications Engineer - Embedded AI will play a key role in developing the customer-facing components of Axelera AI’s Voyager SDK, empowering developers to rapidly evaluate, prototype, and deploy production AI solutions on Axelera AI hardware. You'll work closely with cross-functional teams to deliver high-performance tools that simplify AI deployment for customers.
What You'll Do
- Studying research papers, open-source code repositories and product documentation to deploy new models using the Voyager SDK, and evaluating their performance and accuracy
- Implementing image pre and post processing operators, collaborating with the compiler team to troubleshoot compilation issues, and writing decoders that transform inference tensors to metadata for ease of use by application developers
- Integrating popular industry-standard model frameworks within the SDK to facilitate low-code and no-code deployment of many customer models and datasets
- Defining metadata representations for common types of models (such as bounding boxes and keypoints), creating libraries for evaluating deployed model accuracy on Axelera AI hardware and creating libraries for visually rendering inference results
- Optimizing end-to-end pipeline latency and throughput, root causing any bottlenecks arising on supported hardware platforms, and developing profiling tools to help customers more easily analyse and understand bottlenecks in their own pipelines
- Writing and maintaining comprehensive product documentation and sharing knowledge with team members, promoting best practices and continuous improvement
What We're Looking For
- BS/MS in Computer Science, Electrical Engineering or equivalent work experience in the semiconductor and/or AI industry
- Proven hands-on experience with edge deployment frameworks such as OpenVINO or TensorRT
- Proficiency in AI application development using Python, with extensive experience in ML libraries such as PyTorch and TensorFlow
- Demonstrated ability to build and deploy end-to-end pipelines with quantized models
- Experience in training models using transfer learning to leverage pre-trained models for new tasks
- Strong problem-solving skills and an ability to troubleshoot and debug complex multi-dimensional problems
- Proficient Linux skills
- Practical knowledge of agile development using tools such as Jira, Git and GitHub
- Self-motivated with the ability to thrive in a fast-paced collaborative startup environment
Nice to Have
- Familiarity with model optimisation techniques such as quantization, compression and pruning
- Experience with MLIR and ONNXRuntime
Technical Stack
- Python, PyTorch, TensorFlow, OpenVINO, TensorRT, ONNXRuntime, MLIR, Git, GitHub, Jira, Linux
Team & Environment
- 220+ employees (including 49+ PhDs)
- Applications team within a global R&D organization
Benefits & Compensation
- Flexible working arrangement with office or remote options in Europe
- Opportunity to work from Axelera AI offices in Leuven (Belgium), Amsterdam and Eindhoven (Netherlands), Florence and Milan (Italy), or Bristol (UK)
- Option to work fully remotely from any European country (including the UK)
- Relocation support to Italy (Florence or Milan), Belgium (Leuven), or the Netherlands (Amsterdam or Eindhoven)
- Attractive compensation package
- Pension plan
- Extensive employee insurances
- Option to get company shares
- Open culture supporting creativity and continual innovation
- Collaborative ownership and freedom with responsibility
- Equal opportunity and diversity commitment
- Warm and inclusive environment
Work Mode
- Local-country work mode with locations in Belgium (Leuven), Netherlands (Amsterdam, Eindhoven), Italy (Florence, Milan), United Kingdom (Bristol), and remote within Europe (including UK)
- Flexible working arrangement with office, remote, and relocation options; priority given to candidates based in Bavaria (Munich), Belgium, or Italy
Axelera AI wholeheartedly embraces equal opportunity and holds diversity in the highest regard. The company is committed to cultivating a warm and inclusive environment that empowers and celebrates every member of the team. Applicants from all backgrounds are welcome.
