Machine Learning Engineer
Machine Learning Engineer
Endiatx — Hayward, California (On-site)
About Endiatx
Endiatx is a medical device company developing microrobotic systems for diagnostic use inside the human body. Our flagship platform, PillBot™, is a propeller-driven endoscopic capsule designed to enable real-time visualization of the stomach without traditional endoscopy.
We are a multidisciplinary team of engineers, clinicians, and operators building a new category of medical diagnostics at the intersection of robotics, imaging, and AI.
Role Summary
We are hiring a Machine Learning Engineer to lead development and deployment of our endoscopic image analysis systems. This role owns the full ML lifecycle—from data strategy through production deployment—and contributes directly to how our devices interpret and interact with the physical world.
You will work on computer vision models for gastric imagery, image enhancement under constrained in vivo conditions, and perception systems supporting navigation and localization. You will collaborate closely with hardware, firmware, and simulation teams to integrate ML into a real-world medical device system.
This is a mid-level role with significant ownership and the opportunity to mentor junior contributors.
What You Will Do
- Own the end-to-end ML pipeline: data curation, model development, evaluation, deployment, and monitoring
- Design, train, and validate computer vision models using frameworks such as PyTorch
- Develop image enhancement pipelines
- Optimize models for performance across accuracy, latency, and compute constraints
- Implement neural compression techniques
- Build and maintain dataset infrastructure
- Deploy models into production environments with 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
Required Qualifications
- 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
Nice to Have
- Medical imaging experience (e.g., DICOM)
- C# or Unity familiarity
- SLAM or localization experience
- Robotics or autonomous systems experience
Location
On-site in Hayward, CA. Limited hybrid flexibility.
Compensation
$130,000 – $180,000 base salary, plus equity and benefits.
Work Authorization
Must be authorized to work in the U.S.
Equal Opportunity
Endiatx is an equal opportunity employer.
Accommodations
Reasonable accommodations provided.
Employment Status
At-will employment.