JSG is seeking a Sr. level Machine Learning Engineer (ML) for our client in the Bay Area. This role focuses on developing and optimizing lightweight machine learning models for edge AI applications, particularly in audio processing, and deploying them on embedded platforms like FPGAs and custom ASICs.
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
** Johnson Service Group (JSG) is an Equal Opportunity Employer. JSG provides equal employment opportunities to all applicants and employees without regard to race, color, religion, sex, age, sexual orientation, gender identity, national origin, disability, marital status, protected veteran status, or any other characteristic protected by law.
Join us in Fremont, California, where exciting job opportunities await! Nestled in the heart of the San Francisco Bay Area, Fremont offers a vibrant mix of culture, innovation, and natural beauty. Home to landmarks like the Niles Essanay Silent Film Museum and the Ardenwood Historic Farm, this diverse city boasts a rich arts scene, delicious cuisine, and proximity to major tech companies and Silicon Valley. With easy access to outdoor recreation at the Don Edwards National Wildlife Refuge and the Mission Peak Regional Preserve, Fremont is the perfect place to grow your career and enjoy a high quality of life. Explore our job listings today and discover the charm of Fremont!
Are you sure you want to apply for this job?
Please take a moment to verify your personal information and resume are up-to-date before you apply.
