Machine Learning Engineer

Endiatx; Hayward (CA); On-site; USD 130,000 - 180,000; Full time

Benefits offered by this job

Equity

Benefits

Job summary

Endiatx in Hayward, CA is seeking a Machine Learning Engineer to lead end-to-end development and deployment of endoscopic image analysis systems on-site. You will own the ML lifecycle, design computer vision models with PyTorch, and optimize for accuracy, latency, and compute constraints.

Requirements include 4+ years of ML engineering, production CV model deployments, strong Python and PyTorch skills, and experience with MLOps and CI/CD.

Qualifications

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

Responsibilities

• Own the end-to-end ML lifecycle: data curation, model development, evaluation, deployment, monitoring.
• Design, train, and validate computer vision models using PyTorch.
• Develop image enhancement pipelines for in vivo constraints.
• Optimize model performance for accuracy, latency, and compute constraints.
• Implement neural compression techniques.
• Deploy models into production with CI/CD and monitoring.
• Contribute to localization and pose estimation, perception systems.
• Mentor interns and junior engineers; communicate technical work clearly.

Skills

4+ years ML engineering

Production CV models

Deep learning

Software engineering fundamentals

Communication skills

Image enhancement, compression, or per

Tools

Python

PyTorch

MLOps

CI/CD

Monitoring

Job description

Endiatx is hiring a Machine Learning Engineer to lead end-to-end development and deployment of endoscopic image analysis systems on-site in Hayward, CA.

Responsibilities

• Own the end-to-end machine learning lifecycle, including data curation, model development, evaluation, deployment, and ongoing monitoring
• Design, train, and validate computer vision models using frameworks such as PyTorch
• Develop image enhancement pipelines for in vivo constraints
• Optimize model performance across accuracy, latency, and compute constraints
• Implement neural compression techniques
• Build and maintain dataset infrastructure
• Deploy models into production environments using CI/CD and monitoring
• Contribute to localization and pose estimation
• Support perception systems for navigation and scene understanding
• Collaborate with simulation and visualization teams
• Mentor interns and junior engineers
• Communicate technical work clearly

Requirements

• 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 with image enhancement, compression, or perception systems

Technologies

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

Nice to have

• Medical imaging experience (e.g., DICOM)
• Familiarity with C# or Unity
• Experience with SLAM or localization
• Experience in robotics or autonomous systems

Location

• On-site in Hayward, CA (limited hybrid flexibility)

Compensation

• $130,000 - $180,000 base salary per year
• Equity and benefits included
• Pay range: $130,000.00 - $180,000.00 per year

Work authorization

• Must be authorized to work in the U.S.

Employment status

• At-will employment

Accommodations

• Reasonable accommodations provided