ML Runtime Optimization Engineer - Lead
Applied Intuition
about 1 year ago
Mountain View, CA, USA
Senior / Staff+
H1B Sponsor
Base Salary
$199k - $265k/yr
Responsibilities
- Drive ML performance optimization for ADAS/AD stacks on embedded platforms.
- Provide technical leadership to the ML model performance optimization team.
- Develop strategies to optimize efficiency and latency of model inference.
- Work on model pruning and quantization for memory-constrained platforms.
- Collaborate with ML engineers and software developers on architecture solutions.
- Profile model performance on embedded platforms and identify bottlenecks.
Requirements
- Bachelor's degree in Electrical Engineering, Computer Science, or related field.
- 5+ years of experience with ML accelerators and embedded systems.
- Strong software development skills focused on embedded programming.
- Experience profiling and optimizing model performance on embedded platforms.
- Familiarity with deep learning frameworks like PyTorch and JAX.
Benefits
- Comprehensive health, dental, and vision insurance.
- 401k retirement benefits with employer match.
- Learning and wellness stipends.
- Paid time off.
Tech Stack
PyTorch
Categories
AI & MLEmbedded