Machine Learning Engineer
UnitX, Milpitas, CA - HQ
What you will do
- You will develop new algorithms that consume raw sensor data input to detect defect segmentation and location information on high-resolution images or 3D inputs, with pixel-level precision.
- You will engineer software that runs on the production line 24/7 to efficiently execute our algorithms in real time with each decision made under 20ms.
- You will develop metrics and tools to evaluate various model performance and improve visibility and interpretation of our system.
- You will explore new solutions and push the boundaries of the field including using Stable Diffusion, SAM and build applications for critical missions in the manufacture fields.
What you need
- Bachelor's degree in Computer Science, Mathematics, Physics, or a relevant technical field, or equivalent practical experience. Solid foundational and applied math knowledge.
- 2+ years of experience in building machine learning models for computer vision in production.
- Strong theoretical background and practical experience in Deep Learning, with proficiency in PyTorch or Tensorflow. Excellent Python skills for writing efficient and maintainable solutions in large code bases.
