Skip to content
New Job?Nejo!

The AI Job Search Engine

AXAxelera AI

Senior/Staff Backend Compiler Engineer(m/w/x)

Zürich
Full-timeRemoteSenior
AI/ML

Optimizing in-house DSL and compiler backends for heterogeneous AI hardware at 150-person semiconductor startup. Deep expertise in low-level computation optimization and computer architecture required. Company share options, comprehensive pension and insurance package.

Requirements

  • Master’s, PhD in Computer Science, or related field
  • 5+ years experience in compiler development
  • Understanding of compiler backend concepts
  • Experience optimizing computations at low level
  • Understanding of computer architecture concepts
  • Familiarity with heterogeneous computing architectures
  • Strong Python and good C++ skills
  • Collaborative teamwork skills
  • Experience with modern compiler frameworks
  • Experience with domain-specific languages
  • Experience with RISC-V architecture
  • Experience optimizing deep learning workloads
  • Experience with hardware/software co-design
  • Experience in performance analysis and profiling

Tasks

  • Add new features to the in-house DSL
  • Enhance the existing compiler stack
  • Improve the kernel-developer experience
  • Optimize the compiler backend for heterogeneous hardware
  • Manage compute and memory usage efficiency
  • Optimize data movement across memory hierarchies
  • Evolve the kernel language for internal and external users
  • Design backend optimizations for CPU and NPU
  • Implement memory planning and tiling strategies
  • Apply vectorization and task partitioning techniques
  • Develop concurrency optimizations for compute and memory
  • Contribute to compiler infrastructure and intermediate representations
  • Build and refine compiler optimization passes
  • Analyze performance bottlenecks using profiling and benchmarking
  • Implement performance improvements based on data
  • Provide technical direction and mentor junior engineers
  • Promote engineering best practices within the team
  • Collaborate with a distributed engineering team

Work Experience

5 years

Education

Master's degree

Languages

EnglishBusiness Fluent

Tools & Technologies

PythonC++MLIRLLVMTVMTritoncuTileRISC-VARMx86GPUNPUDSP

Benefits

Competitive Pay

  • Attractive compensation package
  • Company shares option

Retirement Plans

  • Pension plan

Other Benefits

  • Extensive employee insurances

Informal Culture

  • Open culture
  • Equal opportunity environment
  • Inclusive work environment

Startup Environment

  • Collaborative ownership

Purpose-Driven Work

  • Freedom with responsibility
Find the original job posting in its most current version here. Nejo automatically captured this job from the website of Axelera AI and processed the information on Nejo with the help of AI for you. Despite careful analysis, some information may be incomplete or inaccurate. Please always verify all details in the original posting! Content and copyrights of the original posting belong to the advertising company.
Not a perfect match?
100+ Similar Jobs in Zürich
  • Axelera AI

    Staff Engineer - CPU & ISA Architecture(m/w/x)

    Full-timeRemoteExperienced
    Berlin, Zürich, München
  • Axelera AI

    Senior/Staff Applications Engineer - Embedded AI(m/w/x)

    Full-timeRemoteExperienced
    Zürich, München
  • Axelera AI

    Director - Silicon Logical Design(m/w/x)

    Full-timeRemoteSenior
    Zürich
  • Axelera AI

    Principal Engineer - Silicon Logical Design(m/w/x)

    Full-timeRemoteSenior
    Zürich
  • Axelera AI

    Processor Verification Engineer(m/w/x)

    Full-timeRemoteSenior
    Zürich
100+ View all similar jobs

Nejo is an AI – results may be incomplete or contain mistakes