System Engineer

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 and deploy lightweight edge AI models, with emphasis on audio processing. You will optimize models for embedded platforms such as FPGA and custom ASICs, collaborating with hardware teams to productionize solutions.

The role requires deep ML expertise (PyTorch, TensorFlow, JAX) and strong software skills (Python, C/C++), plus a track record of optimizing for latency, power, and size.

Qualifications

• 5+ years of experience in ML, or PhD in related field
• Expertise in edge AI and lightweight model deployment
• Proficiency with ML frameworks such as PyTorch, TensorFlow, JAX

Responsibilities

• Develop, optimize, and deploy lightweight ML models for edge AI, esp. audio
• Implement ML models on embedded platforms (FPGA, ASIC)
• Collaborate with hardware/software teams to productionize ML models
• Research state-of-the-art ML techniques to improve efficiency, latency, and power
• Improve inference efficiency and model compression (quantization, pruning, distillation)
• Provide mentorship to junior engineers and publish findings when applicable

Skills

Edge AI

Model optimization

Python

C/C++

PyTorch

TensorFlow

Education

PhD in Computer Science or Electrical Engineering

Tools

ONNX

TensorRT

TFLite/LiteRT

ncnn

CoreML

Job description

• 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.

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.

Requirements

• 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.

Experience in one or more of the following areas considered a strong plus

• Understanding of ML compiler and runtime design.
• Experience working with tools such as Optimum, ONNX, TensorRT, TFLite/LiteRT, ncnn, or CoreML.
• Familiarity with hardware acceleration techniques.
• Experience in embedded system development.

Salary Range: $110,000 - $250,000 / year

TetraMem celebrates diversity and is committed to creating an inclusive environment for all employees. We are proud to be an Equal Opportunity Employer and welcome applicants from all backgrounds.

Qualified candidates will receive consideration for employment without regard to race, color, religion, creed, sex, gender identity or expression, sexual orientation, national origin, ancestry, age, marital status, medical condition, disability, genetic information, military or veteran status, or any other characteristic protected by applicable federal, state, or local law.

TetraMem is committed to providing reasonable accommodations to qualified applicants with disabilities throughout the recruitment process. Applicants requiring accommodation may contact Human Resources for assistance.