We are seeking a highly skilled Software/ML Engineer to join the Wearable System Architect team and work on power and performance optimization of on device ML.

NO C2C 

The area of focus and main problems to address are:

• Provide ML workload partition definition based on PnP characterization of ML accelerators.
• Provide clear PnP guidelines for ML model architectural exploration
• End-to-End power and performance optimization of AI driven use cases.

RESPONSIBILITIES

Collect power and performance measurement results and traces of ML benchmarks (e.g., MLPerf-Tiny).
Execute ML benchmarks under different on-device configurations, for example:
Execute ML benchmark on different on device ML accelerators.
Execute ML benchmark on slow/external memory and fast/internal memory

For this, the candidate needs to able to compile an existing ML model against different ML accelerators using corresponding ML compilers. The candidate also needs to be familiar with RTOS and Android development and run-time environments.

• Analyze above results to reveal PnP (Power and Performance) characterization of different ML accelerators.

This will eventually lead to the workload partition definition, i.e., which type of ML workload will be more suitable on which ML accelerators.

• Modify ML benchmark models by varying different key ML model parameters.

For example:

Increase # of MACs significantly while keeping memory throughput relatively steady or vice-versa.
For this, the candidate needs to able to modify an existing ML model by changing model parameters (e.g., increasing dimension of CNN layer) or model architecture (e.g., add a Fully Connected layer).

• Analyze above results in order to reveal the relationship between key ML model parameters (e.g., # of MACs) and PnP metrics

This will eventually lead to PnP guideline that can help project PnP metrics based on values of key ML model parameters.

• Collect power and performance traces of AI driven use cases and identify areas of optimization.

QUALIFICATIONS

MINIMUM QUALIFICATIONS

• BS in Computer Science or Computer Engineering
• 2+ consumer product (e.g. Phone, Watch, Glass or other) experience.
• Familiar with RTOS, Android and embedded development environment.
• Familiar with ML development environment.
• BS in Computer Science or Computer Engineering
• 2+ consumer product (e.g. Phone, Watch, Glass or other) experience.
• Familiar with RTOS, Android and embedded development environment.
• Familiar with ML development environment.
• The ideal candidate shall have strong firmware skills and will implement tiny ML on hardware/MCUs.
• Training the ML models including model architectural exploration is outside the scope of this job. But the candidate will need to modify ML models for the purpose of revealing the relationship between PnP metrics and ML model parameters. This will require the candidate having skills to port and compile ML models to run on device.