AI/ML Software Engineer at Flex

Posted in Engineering 3 days ago.

This job brought to you by eQuest

Type: Full-Time
Location: Milpitas, California

Job Description:

Flex is the manufacturing partner of choice that helps a diverse customer base design and build products that improve the world. Through the collective strength of a global workforce across 30 countries and responsible, sustainable operations, Flex delivers technology innovation, supply chain, and manufacturing solutions to various industries and end markets. Create the extraordinatry:


  • Industry: Electrical / Electronic Manufacturing

  • Type: Public Company


  • Headquarters: 2 Changi South Lane, Singapore, 486123

  • Company Size: 10,001+ employees

  • Website:


Please click here to be redirected to the internal career site.

Job Description

The AI/ML Software Engineer will be based in Milpitas, California. We are seeking an adaptable, creative Machine Learning/AI Engineer for a position on the Flex Corporate Innovation engineering team. This is an exciting opportunity to join a growing software system design and applications engineering team.  You will provide subject-matter expertise in AI/ML design as part of a team developing innovative robotics, vision, smart human interfaces, and audio products. You will work on cutting edge products to design and implement innovative and cost-effective, manufacturable product solutions of complex scope in order to support the organization from early prototype development.  You will have the opportunity to develop professionally within a small team and communicate your achievements with the broader Flex organization and mentor junior engineers.

We are looking for someone who demonstrates:

  • Intense collaboration

  • Passionate customer focus

  • Thoughtful, fast, disciplined executor

  • Tenacious commitment to continuous improvement

  • Relentless drive to win

Here is a glimpse of what you’ll do:

  • Design, implement and optimize data processing and AI/ML model using mainstream AI frameworks

  • Prototype AI/ML-based solutions by performing training, validation, and testing

  • Design develop deep Learning models like CNN, RNN, and NLP (BERT) for solving various business use cases for, radar, vision, and audio systems like presence, small object, and voice pattern detection.

  • Research and develop new algorithms such as video stream object detection and tracking, radar gesture recognition, environment noise cancellation, dynamic audio filters and targeting edge devices.

  • Design state-of-the-art sensor systems & human-machine interfaces for a variety of products ranging from smart devices, robotics and advance audio systems.

  • Mentor other developers and researchers.

  • Collaborate closely with team members of various department in the Innovation Center to deliver proof of concept

  • Maintain rigorous software writing practices, including documentation and versioning

  • Collaborate with product teams to deliver core AI model APIs

  • Maintain up to date knowledge and skill about AI technologies and related software tools

  • Should be comfortable working on open-ended problems that require sustained effort to both specify and solve.

  • A multi-disciplinary AI/ML skill set is a must. Familiarity and knowledge of edge-based ML accelerators is highly desirable

Here is some of what you’ll need (required):

  • MS degree in Computer Science or equivalent with minimum 3 years of experience in AI/ML algorithm, model development, and deployment on edge compute devices.

  • Excellent familiarity with OpenCV and libraries such as Numpy, Scipy, PyTorch, and DL frameworks like TensorFlow, Keras, and MLOps.

  • Deep understanding of vision and audio processing with AI/ML.

  • Experience implementing algorithms in Object detection and tracking (ex: MobileNetSSD, KCF tracker), noise cancellation, or NLP.

  • Experience in Computer Vision; Pattern Recognition; Predictive Modeling and Decision Support; Reduced Order Modeling.

  • Familiarity with audio-based AI/ML techniques

  • Some familiarity with cognitive agents (intelligent bots).

  • General knowledge of edge compute requirements and SoCs targeting edge inferencing.

  • A strong commitment to writing robust, high-quality code using C++ and Python.

  • Demonstrated ability to innovate and deliver product quality results

  • Familiarity with machine learning accelerators from SoC vendors is desirable.

  • Some experience in targeting CNN and RNN models for embedded SoCs.

  • A good understanding of inference in real-time systems.

  • Display solid interpersonal skills including strong written and verbal communication capabilities, both locally and with remote design teams

Here are a few of our preferred experiences:

  • Experience leading design projects that span multiple sites and engineering teams

  • Understanding of intellectual property, patent, and trade secret business processes

  • Familiarity with model compression techniques like Knowledge Distillation.

Here are a few examples of what you’ll get for the great work you provide:

  • Full range of medical benefits, dental, vision

  • Life Insurance

  • Matching 401K

  • PTO

  • Tuition Reimbursement

  • Employee discounts at local retailers

Job Category

Design & Technology


Is Sponsorship Available?


Flex does not accept unsolicited resumes from headhunters, recruitment agencies or fee based recruitment services. Flex is an Equal Opportunity Employer and employment selection decisions are based on merit, qualifications, and abilities. Flex does not discriminate in employment opportunities or practices based on: age, race, religion, color, sex, national origin, marital status, sexual orientation, gender identity, veteran status, disability, pregnancy status or any other status protected by law. Flex provides reasonable accommodation so that qualified applicants with a disability may participate in the selection process. Please advise us of any accommodations you request to express interest in a position by e-mailing: Please state your request for assistance in your message. Only reasonable accommodation requests related to applying for a specific position within Flex will be reviewed at the e-mail address. Flex will contact you if it is determined that your background is a match to the required skills required for this position. Thank you for considering a career with Flex.

The information we collect:

We may collect personal information that you choose to submit to us through the Website or otherwise provide to us. This may include your contact details; information provided in online questionnaires, feedback forms, or applications for employment; and information you provide such as CV/Resume. Your details will be provided to the entity you are applying for a job with. We will use your information for legitimate business purposes such as responding to comments or queries or answering questions; progressing applications for employment; allowing you to choose to share web content with others or; where you represent one of our customers or suppliers, administering the business relationship with that customer or supplier. We will process your data in accordance with our Recruitment Privacy Notice.

If you have any queries about the processing of your data, please contact:


Global Data Privacy Officer:

Email Address: