(Sr./Staff) Software Engineer, Edge AI - Toolchain

OMNIVISION TECHNOLOGIES SINGAPORE PTE. LTD.; Singapore; On-site; SGD 80,000 - 130,000; Full time

Job summary

A leading tech firm is seeking Senior/Staff Software Engineers for their Edge AI & Systems team in Singapore. The role involves developing and enhancing toolchain features that enable AI model optimization on Neural Processing Units. Candidates should possess strong skills in Python, compilers, and have a robust educational background in Computer Science or related fields, with significant professional experience.

Qualifications

• Ph.D., MS, or BS in Computer Science or Computer Engineering.
• Minimum 5 years commercial development experience, 3 years in compilers.

Responsibilities

• Develop and specify toolchain features for NPU architectures.
• Enhance and support existing toolchain features.
• Improve productivity through automation.

Skills

Strong Python product development

Understanding of compilers

Good communication and collaboration

CI/CD knowledge via tools

Education

Ph.D./MS/BS in Computer Science or Engineering

Tools

Confluence

Git

JIRA

Job description

Job Description

We are looking for Senior/Staff Software Engineers to grow our Edge AI & Systems software team to the next level. This team works within the larger ASIC team to architect and enhance Neural Network solutions in the Video Processor roadmap including these current and older products:

• https://www.ovt.com/products/oax8000/
• https://www.ovt.com/products/oax4600/
• https://www.ovt.com/products/oa7100/

This position works in a smaller team that specifies, develops and delivers the toolchain that allows our customers and partners to develop or port AI models to run optimally on our Neural Processing Units.

Responsibilities

• Specify and develop new toolchain features for existing and new NPU architectures, working with the inference engine sub-team
• Enhance, maintain and support existing toolchain features, and modify model network layers for NPU optimality, working with the application engineering teams and sometimes customers/partners.
• Constantly improve productivity through automation in all areas

Skills

• Strong Python class based product development skills based on knowledge of design patterns
• Strong understanding of all aspects of a compiler and related development and debugging technologies, code generation, program analysis, and pipeline optimizations
• Good communication, collaboration, task estimation & reporting, CI/CD via tools such as Confluence, Git, JIRA etc.
• Familiarity with the following is a plus:
• Deep Learning compilers like TVM
• Deep Learning framework - PyTorch
• Eclipse development for toolchain frontend
• CPU/DSP/GPU/NPU architectures

Requirements

1. Ph.D./MS/BS in Computer Science or Computer Engineering
2. At least 5 years of commercial development experience of which at least 3 are in compiler-related areas