Embedded AI Engineer, On-Device Models

Deepgram

• $130K — $160K *
US-AnywhereRemote in United States
Consumer Technology
Less than 5 years of experience
Job Overview by Ladders

Qualifications

  • 5+ years of experience in embedded systems or edge AI development.
  • Proficiency in C, C++, and/or Rust for performance-critical programming.
  • Hands-on experience with model optimization techniques such as quantization and pruning.
  • Familiarity with edge inference runtimes like ONNX Runtime and TensorRT.
  • Strong understanding of hardware-software interaction and inference performance optimization.

Responsibilities

  • Optimize and deploy Deepgram’s speech models on embedded consumer hardware.
  • Write performance-critical runtime code for various embedded operating systems.
  • Integrate edge inference runtimes and optimize for diverse processor architectures.
  • Build deployment pipelines and models for on-device telemetry and updates.
  • Establish benchmarking and validation protocols for hardware performance.

Benefits

  • Work in a cutting-edge field at the intersection of AI and hardware.
  • Join a team that values practical applications over theoretical research.
  • Opportunity to shape the future of voice AI in consumer devices.
Full Job Description
About the Role

Deepgram's speech models are among the fastest and most accurate in the world, and we have deep machinery for running them on NVIDIA GPUs. Our customers need them on everything else: non-NVIDIA accelerators, embedded SoCs, mobile application processors, DSPs and NPUs, and purpose-built devices with tight memory, compute, thermal, and power budgets. When a target platform's standard kernels and runtime can't run a Deepgram model well enough, someone has to go below them. That is this role.

As an Embedded AI Engineer on the Partner Platform Engineering team, you work at the lowest layer of our edge stack. You write and optimize custom kernels and operators for specific hardware, collapse models onto device-specific execution units, and do the target-side quantization and assembly-level tuning that standard toolchains can't. You hand what you build up to Applied ML Engineers, who fit Deepgram models to your kernels. Your work is what makes a new hardware platform viable for Deepgram at all.

This role is a great fit for a senior embedded engineer who has spent their career close to the metal and wants to point that at speech AI, or a staff-level engineer who wants to define how Deepgram's models get onto new silicon. We'll set the level to your experience.

What You'll Do
  • Write and optimize custom kernels and operators (C, C++, Rust, and platform assembly or intrinsics) for non-NVIDIA accelerators, embedded SoCs, DSPs, and NPUs where the vendor's standard operator set is insufficient for Deepgram models.
  • Own target-side optimization: collapse models onto device execution units through quantization, operator fusion, memory layout, and architecture-specific compilation to meet latency, memory, power, and thermal budgets.
  • Integrate with vendor NPU/DSP toolchains and edge inference runtimes, and extend them with custom operators when the graph doesn't map cleanly.
  • Deliver kernels and runtime components as reusable building blocks that Applied ML Engineers can target when adapting models, with clear interfaces and documented constraints.
  • Build performance-critical runtime code for embedded environments, including embedded Linux, bare-metal, and RTOS targets.
  • Establish per-platform benchmarking and validation for latency, accuracy, power, memory footprint, and utilization, and catch regressions before they ship.
  • Partner with silicon and platform vendors on SDK integration and low-level performance tuning for new chipsets and reference platforms.
  • Feed hardware constraints back to Applied ML and Research so model designs are easier to land on constrained targets.


You'll Love This Role If You
  • Find deep satisfaction in making a large model run on hardware that was never meant to run it, and still hitting accuracy and latency targets.
  • Reach for the profiler and the ISA manual before you reach for a bigger chip.
  • Would rather write the kernel than wait for the vendor to ship it.
  • Care about the details that don't show up in a cloud benchmark: cold start, power draw, thermals, memory fragmentation, cache behavior.
  • Prefer hard, constrained, ship-it problems over open-ended research.
  • Care about the details that don't show up in a cloud benchmark: cold-start time, power draw, thermals, and memory fragmentation.


It's Important To Us That You Have
  • Experience delivering production systems on resource-constrained hardware - embedded systems, mobile, edge AI, or small low-power devices.
  • Strong proficiency in C, C++, and/or Rust, with experience writing performance-critical code for constrained environments.
  • Hands-on experience with model optimization for on-device deployment, including quantization, pruning, knowledge distillation, or architecture-specific compilation.
  • Familiarity with edge inference runtimes (e.g., ONNX Runtime, TensorRT, TFLite, ExecuTorch) and/or vendor-specific NPU/DSP toolchains.
  • A strong understanding of hardware-software interaction - CPU/GPU/NPU/DSP architectures, memory hierarchies, fixed-point/integer arithmetic, and power management - and how they affect inference performance.
  • Experience working close to the metal: bare-metal or RTOS environments (e.g., FreeRTOS, Zephyr), embedded Linux, or microcontroller and edge SoC development.
  • Strong communication skills and a builder mindset - you can scope an ambiguous optimization problem, drive it to a measurable result, and explain the tradeoffs clearly.


It Would Be Great if You Had
  • Experience with real-time audio processing on embedded platforms - DSP pipelines, audio codec optimization, wake-word or always-on listening, or streaming inference on microcontrollers and edge SoCs.
  • Depth in ML optimization techniques - custom quantization schemes, mixed-precision inference, or neural architecture search for edge targets.
  • Background in hardware evaluation and benchmarking - systematically comparing accelerators, SoCs, or GPUs for specific workload profiles.
  • Experience shipping AI features in consumer products at scale, and the instinct for what "production quality" means on a battery-powered device.
  • Familiarity with model compilation and optimization toolchains and their tradeoffs across hardware targets.
  • Experience with secure, robust on-device deployment practices - code signing, encrypted model storage, and safe update mechanisms.

Similar Jobs

More Jobs at Deepgram

More Consumer Technology Jobs

Find similar Embedded AI Engineer, On-Device Models jobs: