Edge AI System Engineer — Lightweight ML on Embedded

TetraMem - Accelerate The World; San Jose (CA); On-site; USD 110,000 - 250,000; Full time

Job summary

TetraMem - Accelerate The World is seeking a senior ML engineer to design and optimize edge AI models for audio processing on embedded platforms, including FPGA and ASIC. You will collaborate with hardware and software teams to deploy production-ready ML solutions.

The role emphasizes research-driven model efficiency, latency reduction, and power optimization, with opportunities to mentor junior engineers and publish findings.

Qualifications

• 5+ years of experience or PhD in Computer Science, Electrical Engineering, or related fields.
• Strong experience in machine learning, with a focus on edge AI and lightweight model deployment.
• Expertise in ML frameworks such as PyTorch, TensorFlow, JAX.
• Proficiency in programming languages such as C/C++, Python, and experience with ML model optimization.

Responsibilities

• Develop, optimize, and deploy lightweight machine learning models for edge AI applications, particularly for audio processing.
• Implement and optimize ML models on embedded platforms, including FPGA and custom ASIC solutions.
• Work closely with hardware and software teams to integrate ML models into production systems.
• Research and implement state-of-the-art ML techniques to enhance model efficiency, latency, and power consumption for embedded AI applications.
• Improve inference efficiency and model compression techniques, including quantization, pruning, and knowledge distillation.
• Collaborate with cross-functional teams to drive innovation and contribute to the overall system architecture.
• Provide technical leadership and mentorship to junior engineers.
• Publish research findings, present at conferences, and contribute to open-source projects when applicable.

Skills

Edge AI

PyTorch

TensorFlow

JAX

C/C++

Python

Model optimization

Education

PhD or MS in Computer Science or Electrical Engineering

Tools

TensorRT

ONNX

TFLite/LiteRT

ncnn

CoreML

Job description

TetraMem - Accelerate The World is seeking a senior ML engineer to design and optimize edge AI models for audio processing on embedded platforms, including FPGA and ASIC. You will collaborate with hardware and software teams to deploy production-ready ML solutions.

The role emphasizes research-driven model efficiency, latency reduction, and power optimization, with opportunities to mentor junior engineers and publish findings.