Skip to content

Senior Machine Learning Engineer

On-site
  • Santa Clara, California, United States
$180,000 - $220,000 per yearEngineering

Job description

Job Title:

Senior Machine Learning Engineer


About Us:
UnitX is building the world’s best robotics product to accelerate human productivity in manufacturing. UnitX is a fast-moving startup with a team from Stanford and Google. Since inception, UnitX has shipped 500+ mission-critical systems across 61 of the world's leading manufacturers' production lines. Every year, $2.8B dollar worth of products (think EV batteries) go through UnitX AI inspection system to ensure quality.


Join us for the rare opportunity to work on computer-vision-driven products that go beyond the state of the art and that improve global manufacturing efficiency.

Job requirements

What You'll 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 to efficiently execute our algorithms in real time with each decision 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 fields.
Who You Are:
    • Bachelor's degree in Computer Science, Mathematics, Physics, or a relevant technical field, or equivalent practical experience. Solid foundational and applied math knowledge.
    • 5+ years of work experience, with at least 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.
    • 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.
    Preferred Qualifications
    • Experience with image generation with diffusion models and other generative models.
    • Experience with model inference optimization
    • Experience with non-ML CV algorithms

Compensation & Benefits

  • $180,000 - 220,000 + equity
  • Full Medical, Dental, Vision, 401k
  • Unlimited PTO
  • Daily meals provided

or