Machine Learning Engineer

Endiatx is seeking a Machine Learning Engineer to lead the end-to-end development and deployment of endoscopic image analysis systems. This role owns the full machine learning lifecycle, from data strategy through production deployment and ongoing monitoring. You will work on computer vision models for gastric imagery, image enhancement under in vivo constraints, and perception systems that support navigation and localization, collaborating closely with simulation and visualization teams along the way.

On-site in Hayward, CA with limited hybrid flexibility. This is a U.S. work authorization required position, offered as at-will employment. Reasonable accommodations are provided.

Compensation: $130,000 to $180,000 base salary per year, plus equity and benefits.

What you’ll do

• Own the end-to-end ML pipeline, including data curation, model development, evaluation, deployment, and monitoring
• Design, train, and validate computer vision models using PyTorch and related deep learning workflows
• Develop image enhancement pipelines and improve model outputs under in vivo constraints
• Optimize models across accuracy, latency, and compute constraints for production readiness
• Implement neural compression techniques to meet performance requirements
• Build and maintain dataset infrastructure to support reliable training and iteration
• Deploy models into production using CI/CD and establish monitoring for model performance
• Contribute to localization and pose estimation and support perception systems for navigation and scene understanding
• Collaborate with simulation and visualization teams
• Mentor interns and junior engineers
• Communicate technical work clearly with cross-functional partners

What you’ll need

• 4+ years of ML engineering experience (or equivalent)
• Experience deploying production computer vision models
• Strong foundation in deep learning
• Proficiency in Python and PyTorch
• Experience with MLOps tools
• Strong software engineering fundamentals
• Strong communication skills
• Experience in image enhancement, compression, or perception systems

Technology stack

• Python
• PyTorch
• MLOps
• CI/CD
• CI/CD and monitoring

Nice to have

• Medical imaging experience (for example, DICOM)
• Familiarity with C# or Unity
• Experience with SLAM or localization
• Background in robotics or autonomous systems

Work authorization: Must be authorized to work in the U.S.