Machine Learning Engineer

The role

You will develop algorithms that consume raw sensor data to detect defect segmentation and location on high-resolution images or 3D inputs. You will engineer software for production lines to execute algorithms in real time and develop tools to evaluate model performance.

What you will do

• Develop new algorithms that consume raw sensor data to detect defect segmentation and location information on high-resolution images or 3D inputs
• Engineer software that runs on the production line 24/7 to execute algorithms in real time
• Develop metrics and tools to evaluate model performance and improve system visibility
• Explore new solutions including using Stable Diffusion and SAM

Required qualifications

• 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
• Proficiency in PyTorch or Tensorflow
• Excellent Python skills for writing efficient and maintainable solutions in large code bases
• Strong communication and decision-making skills
• Ability to explain the rationale behind experiments and make decisions to balance exploration and exploitation
• Resilience in uncertain and complex environments

Nice to have

• Master’s degree in Computer Science, Mathematics, Physics, or a relevant technical field
• Experience with model inference optimization
• Experience with non-ML CV algorithms

Before you apply

• Employment is contingent upon successful completion of applicable pre-employment screening requirements

About the company

UnitX builds physical AI systems to automate repetitive visual tasks in factories. The company has deployed 1,000+ mission-critical systems across 190+ manufacturers' production lines.