Machine Learning Engineer, Edge AI

Job Description

We are looking for a Principal Firmware Engineer to lead the integration and control of our next‑generation AI accelerators. As our products evolve to include dedicated neural network hardware, the challenge shifts from pure algorithm implementation to complex hardware orchestration. This role is about more than just writing kernels; it’s about defining the firmware layer that sits between high‑level AI frameworks and our custom silicon. You will be responsible for how our DSPs manage, schedule, and feed data to these accelerators. We need a veteran who can assess a PyTorch model and determine the best way to tile memory, manage DMA transfers, and synchronize processing to ensure we hit our ultra‑low‑power targets while maximizing throughput. You will also be the primary technical voice influencing our future hardware specs to ensure our accelerators are actually firmware‑friendly.

Responsibilities

• Architect Accelerator Orchestration: Define and implement the firmware control logic, drivers, and scheduling required to manage dedicated neural network accelerators from the DSP/MCU.
• Bridge the AI Flow: Develop the toolchain strategy that translates PyTorch/TensorFlow models into optimized hardware‑specific instructions and memory layouts.
• Optimize Data Movement: Design efficient DMA and memory tiling strategies to minimize power consumption and latency when moving data between memory and AI hardware blocks.
• Influence Silicon Design: Serve as the lead firmware stakeholder in hardware/software co‑design, ensuring that future hardware accelerators have the right registers, interrupts, and memory interfaces for efficient control.
• Customer Technical Interface: Act as the "deep‑tech" lead for customers, providing the architectural guidance they need to deploy sophisticated models on our specific hardware/firmware stack.
• Performance Bottleneck Analysis: Identify and resolve performance gaps in the interaction between the DSP and the accelerator, moving beyond "it works" to "it's optimal."
• Technical Mentorship: Lead the broader firmware team in modern AI deployment techniques, setting the standard for production‑quality, hardware‑abstracted code.

Must‑Haves

• 15+ years of firmware development experience, with a heavy focus on hardware/software co‑design and controlling custom accelerators or NPUs.
• Deep Hardware Interfacing Knowledge: Expert‑level understanding of DMA, memory‑mapped I/O, interrupts, and synchronization between CPU/DSP and specialized hardware blocks.
• AI Framework Literacy: Strong grasp of the PyTorch/TensorFlow ecosystem and how high‑level graphs are lowered and quantized for embedded deployment.
• Bare‑Metal Mastery: Expert‑level C and a "thinking in hardware" mindset; you should be comfortable reading a hardware spec and identifying firmware bottlenecks before the silicon is even made.
• Resource‑Constrained Optimization: Proven experience squeezing performance out of systems with very limited memory (KB) and strict power budgets.
• Communication: Ability to explain complex hardware‑to‑firmware trade‑offs to senior leadership and provide technical leadership across cross‑functional teams.

Nice‑to‑Haves

• Experience building or using AI compilers or toolchains (e.g., TVM, MLIR, or custom IRs).
• Background in ultra‑low‑power DSPs (audiology, wearables, or medical devices).
• Experience with ARM Cortex‑M or proprietary DSP architectures and their specific bus protocols.
• Knowledge of fixed‑point arithmetic and its impact on neural network performance and power.

About Us

onsemi (Nasdaq: ON) is driving disruptive innovations to help build a better future. With a focus on automotive and industrial end‑markets, the company is accelerating change in megatrends such as vehicle electrification and safety, sustainable energy grids, industrial automation, and 5G and cloud infrastructure. With a highly differentiated and innovative product portfolio, onsemi creates intelligent power and sensing technologies that solve the world’s most complex challenges and leads the way in creating a safer, cleaner, and smarter world.

More details about our company benefits can be found here: https://www.onsemi.com/careers/career-benefits

About The Team

We are committed to sourcing, attracting, and hiring high‑performance innovators, while providing all candidates a positive recruitment experience that builds our brand as a great place to work.

onsemi is an Equal Opportunity Employer. All qualified applicants will receive consideration for employment without regard to race, ethnicity, color, religion, ancestry, national origin, age, marital status, pregnancy, sex, sexual orientation, physical or mental disability, medical condition, genetic information, military or veteran status, gender identity, gender expression, or any other protected category under applicable federal, state, or local laws.

If you are an individual with a disability and require a reasonable accommodation to complete any part of the application process, or are limited in the ability or unable to access or use this online application process and need an alternative method for applying, you may contact Talent.acquisition@onsemi.com for assistance.

#J-18808-Ljbffr