Software Engineer: AI Software Stack (Fixed-Term)

About Us: At Huawei Technologies Switzerland AG, we are a leading technology firm dedicated to developing cutting-edge solutions that redefine industry standards and push technological boundaries. Our core focus is on creating advanced computing architectures that can efficiently support and enhance the performance of artificial intelligence systems. We believe in innovation as a driving force for improvement and are committed to achieving excellence in all areas of research and development.

Job Summary:

We are seeking a highly skilled and motivated Software Engineer to join our team focusing on the development and optimization of AI software stacks and compilation techniques. The successful candidate will have substantial experience with state-of-the-art frameworks and be instrumental in advancing our research ideas. Especially successful candidates may be considered for permanent employment.

Key Responsibilities:

• Contribute to cutting-edge research in AI software stacks, focusing on programmability through Python and PyTorch, bridged into accelerator programming frameworks and compiler technologies;
• Collaborate with cross-functional teams consisting of researchers at different levels of the software stack;
• Write high-quality and high-performance code according to the highest levels of coding standards;
• Ensure new code is integrated with established CI pipelines and conduct code reviews;
• Stay updated on related industry trends and advancements in AI and compiler technologies, and their implications and applications to our research endeavours.
  
  

Qualifications:

• Master or PhD students in Computer Science, Electrical Engineering, or a related field;
• Experience with Python bindings to C++, PyTorch, and the PyTorch / Torch eco-system;
• Strong understanding of programming model, compiler design, program optimization, and parallel processing;
• Excellent problem-solving skills and ability to work in a fast-paced, collaborative environment;
• Strong communication skills with a track record of, and being comfortable with, presenting findings to diverse audiences.
  
  

Preferred Qualifications:

• Recent experience modifying state-of-the-art AI software stacks, e.g. compilers, framework, or system software, is a plus;
• In-depth experience with one or more of Triton, MLIR, TVM, tile_lang, C++20 (or later), etc., is a plus;
• Familiarity with hardware architectures and their impact on AI model performance;
• Contributions to open-source projects or active involvement in the AI research community.
  
  

What We Offer:

• Competitive salary and benefits package;
• Access to state-of-the-art technology and tools;
• Opportunities for professional growth and development;
• Interact with a large pool of expert researchers in a wide variety of areas; and
• Last but certainly not least: be part of innovative projects that make a difference.