Senior Edge AI Engineer — Embedded ML & Audio

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

Job summary

TetraMem is seeking an experienced ML Engineer to develop, optimize, and deploy lightweight models for edge AI, with a focus on audio processing and embedded platforms. You will collaborate with hardware and software teams to integrate models into production and improve efficiency and power consumption.

The role requires 5+ years of ML expertise or a PhD in CS/EE, with strong PyTorch/TF/JAX skills and proficiency in C/C++ and Python.

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.
• Ability to work independently and collaboratively in a fast-paced startup environment.
• Ability to provide mentorship, technical guidance, and career development support to junior engineers and interns.

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

Mentorship

Team collaboration

Education

PhD in CS/EE or related fields

Tools

ONNX

TensorRT

TFLite/LiteRT

ncnn

CoreML

Optimum

ML compiler/runtime

Job description

TetraMem is seeking an experienced ML Engineer to develop, optimize, and deploy lightweight models for edge AI, with a focus on audio processing and embedded platforms. You will collaborate with hardware and software teams to integrate models into production and improve efficiency and power consumption.

The role requires 5+ years of ML expertise or a PhD in CS/EE, with strong PyTorch/TF/JAX skills and proficiency in C/C++ and Python.