Skip to content
Applied IntuitionApplied IntuitionSunnyvale, CA

ML Runtime Optimization Engineer

Optimizes ML models for performance on embedded compute platforms in ADAS/AD stacks, focusing on inference efficiency, pruning, quantization, and profiling across GPU/CPU/SoC architectures. Requires 3+ years experience with deep learning frameworks and embedded systems.

159k – 199k/yr
On-site3+ YOEML Engineering

About the role

Responsibilities

  • Drive ML performance optimization on multiple technologies for on-road and off-road ADAS / AD stacks targeting deployment on a variety of embedded compute platforms
  • Develop compute usage strategies to optimize efficiency and latency of model inference for compute boards selected by our customers
  • Work on model pruning and quantization, and support deployment on memory constrained platforms
  • Collaborate closely with ML engineers and software developers on technical efforts to find and optimize efficient model architecture solutions
  • Set up methodologies to profile the model performance on target embedded compute platforms and identify performance bottlenecks as part of stack integration

Requirements

  • Bachelors in Electrical Engineering or Computer Science, OR B.Sc. in Computer Science, Mathematics, Physics or a related field
  • 3+ years of experience with ML accelerators, GPU, CPU, SoC architecture and micro-architecture
  • Strong software development skills with the focus on embedded programming
  • Experience profiling and optimizing model performance on embedded compute platforms
  • Experience in working with deep learning frameworks (e.g., PyTorch, JAX, ONNX, etc.)

Nice to Have

  • M.Sc or PhD in a ML related area
  • Built an ML optimization framework from scratch before
  • Deployed ML solutions to embedded chips for real time robotics applications

Compensation

  • Base salary range: $159,053 - $199,295 USD annually
  • Equity, comprehensive health, dental, vision, life and disability insurance, 401k with employer match, learning and wellness stipends, paid time off

Skills

PyTorchJAXOnnxTensorRTCUDAXlaTritonGPUMl AcceleratorsEmbedded Programming

Similar roles

ML Engineering jobs
Pindrop

Research Scientist II

PindropUnited States

Research Scientist II building and improving fraud risk models and scam detection systems using audio, behavioral, and metadata signals. Requires an advanced degree and 3+ years of applied ML experience with Python and modern ML frameworks.

160k – 185k/yr
Remote3+ YOEML Engineering
Snowflake

AI Engineer - Database Engineering

SnowflakeMenlo Park, CA

As an AI Engineer, you will own the full AI engineering lifecycle, from design to optimization, for Snowflake Database Engineering products. You will build agentic workflows, coding harnesses, and evaluation pipelines, working with a high-powered engineering team.

160k – 230k/yr
On-site5+ YOEML Engineering
Snowflake

Software Engineer, Cortex AI Infrastructure

SnowflakeMenlo Park, CA

Build and scale backend infrastructure powering agentic AI products including orchestration engines, RAG systems, evals infrastructure, and production AI workflows. Requires 4+ years distributed systems experience and deep Python plus Go/Java proficiency.

160k – 225k/yr
On-site4+ YOEML Engineering
Mach9

ML Infrastructure Engineer

Mach9San Francisco, CA

Builds and maintains ML infrastructure for training pipelines handling massive 3D data and real-time inference serving integrated with CAD software. Requires 3+ years experience with Python, PyTorch, ML orchestration tools, data versioning, and inference optimization.

160k – 200k/yr
On-site3+ YOEML Engineering
Deepgram

ML Ops Infrastructure Engineer

DeepgramCalifornia

Build and maintain ML infrastructure pipelines to deploy research models to production at scale, including CI/CD, A/B testing, monitoring, and optimization for low-latency voice AI serving. Requires 4+ years MLOps experience with Python, Docker, Kubernetes.

160k – 220k/yr
Remote4+ YOEML Engineering