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

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

Job summary

A leading tech company in Singapore is looking for a Senior/Staff Software Engineer to develop and enhance toolchains for AI models running on Neural Processing Units. Candidates should have strong Python skills and an understanding of compiler technologies. The role involves specifying new features, maintaining existing toolchain functionalities, and automating processes. A Ph.D./MS/BS in Computer Engineering or a related field, along with at least 5 years of development experience, is required. This position promotes collaboration within a small team.

Qualifications

• At least 5 years of commercial development experience, with 3 in compiler-related areas.

Responsibilities

• Specify and develop new toolchain features for NPU architectures.
• Enhance and support existing toolchain features.
• Improve productivity through automation.

Skills

Strong Python class based product development skills

Understanding of compiler and related technologies

Good communication and collaboration skills

Education

Ph.D./MS/BS in Computer Science or Computer 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