# ML Runtime Optimization Engineer

**Company:** [Applied Intuition](https://hotfix.jobs/companies/applied-intuition)
**Location:** Sunnyvale, CA
**Role:** ML Engineering
**Salary:** $159k – $199k/yr
**Experience:** 3+ years
**Skills:** PyTorch, JAX, Onnx, TensorRT, CUDA, Xla, Triton, GPU, Ml Accelerators, Embedded Programming
**Posted:** 2026-04-27

> 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.

## Job Description

## 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

## Similar roles

- [Research Scientist II](https://hotfix.jobs/jobs/4ac4da1a-a497-413d-b8fa-6b3b58cef7e1) - Pindrop - Remote - $160k – $185k/yr
- [AI Engineer - Database Engineering](https://hotfix.jobs/jobs/e2167614-4d69-4c69-bb69-40e1c6ba4571) - Snowflake - Menlo Park, CA - $160k – $230k/yr
- [Software Engineer, Cortex AI Infrastructure](https://hotfix.jobs/jobs/cfcfa4ed-3661-45be-8d97-71e231e17de7) - Snowflake - Menlo Park, CA - $160k – $225k/yr
- [ML Infrastructure Engineer](https://hotfix.jobs/jobs/ad8cfc6e-fc74-4f2a-ad7c-a2f5127c6461) - Mach9 - San Francisco, CA - $160k – $200k/yr
- [ML Ops Infrastructure Engineer](https://hotfix.jobs/jobs/3e53824c-d6f1-4156-892e-cca11a6a394d) - Deepgram - Remote - $160k – $220k/yr

**Apply:** https://hotfix.jobs/jobs/7fea6f36-98f8-49a6-9203-6258a8f77181
**Canonical:** https://hotfix.jobs/jobs/7fea6f36-98f8-49a6-9203-6258a8f77181