Sign upLog in
JobsJobs in CaliforniaJobs in San Jose, CAIT Jobs in San Jose, CAMachine Learning Jobs in San Jose, CA

Mid-Level Machine Learning Engineer

TetraMem

San Jose, CA

Apply
JOB DETAILS
SALARY
$110,000–$300,000 Per Year
SKILLS
ASIC (Application Specific Integrated Circuit), Artificial Intelligence (AI), C Programming Language, C++ Programming Language, Computer Science, Computer Skills, Conferences, Cross-Functional, Electrical Engineering, Embedded Systems, FPGA, JAX (Java API for XML), Machine Learning, Mentoring, Open Source, Production Systems, Programming Languages, Python Programming/Scripting Language, Startup, System Architecture, Technical Leadership
LOCATION
San Jose, CA
POSTED
Today
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.
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 - $300,000 / year

About the Company

T

TetraMem

Resume Resources

Free Resume TemplatesFree Resume Builder