Senior System Engineer: Edge AI & Embedded ML Lead

TETRAMEM INC; San Jose (CA); On-site; USD 110,000 - 250,000; Full time

Job summary

TetraMem seeks a seasoned ML Engineer to develop lightweight edge AI models and optimize deployments on embedded platforms. You will lead research, collaborate with hardware and software teams, and push ML innovations for low-latency, power-efficient AI in edge devices and data centers.

Join a fast-moving startup environment where you mentor junior engineers, publish findings, and contribute to open-source projects while advancing state-of-the-art ML techniques for energy-efficient AI.

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 ML 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 deployment

Lightweight models

Mentorship

Team collaboration

C/C++

Python

ML model optimization

Education

PhD in Computer Science or Electrical Engineering

Tools

PyTorch

TensorFlow

JAX

ONNX

Job description

TetraMem seeks a seasoned ML Engineer to develop lightweight edge AI models and optimize deployments on embedded platforms. You will lead research, collaborate with hardware and software teams, and push ML innovations for low-latency, power-efficient AI in edge devices and data centers.

Join a fast-moving startup environment where you mentor junior engineers, publish findings, and contribute to open-source projects while advancing state-of-the-art ML techniques for energy-efficient AI.