Machine Learning Engineer
San Jose, CA
PROCEPT BioRobotics
PROCEPT BioRobotics offers advanced robotic treatments, empowering menâs health and enabling surgeons with cutting-edge precision and technology. Learn more.The Opportunity That Awaits YouAre you ready to revolutionize treatment for men suffering from BPH (Benign Prostatic Hyperplasia)? At PROCEPT, we are building the next generation of intelligent surgical systemsâand you could be a key part of that transformation.
As a Machine Learning Engineer, you will design, develop, and deploy advanced machine learning models that power real-time image-guided robotics. Your work will directly impact surgical decision-making, automate clinical workflows, and enhance patient outcomes.You'll collaborate with top engineers, clinicians, and product leaders to bring AI to life in the operating room. This is your opportunity to join a mission-driven team where innovation mattersâand where your code can save lives.
What You'll DoMachine Learning & Computer Vision-Design, train, and optimize ML models for tasks such as segmentation, object detection, classification, and image enhancement.-Leverage deep learning architectures including CNNs, transformers, and GANs to solve vision challenges in real-time surgical applications.-Apply techniques such as YOLO, Faster R-CNN, U-Net, and ResNet to process high-resolution ultrasound and MRI data. -Develop and integrate AI algorithms for surgical video, multi-modality image fusion, and 3D reconstruction.-Design, train, and fine-tune foundation models such as DINO, SAM, or MAL for computer vision and data scaling.-Build and evaluate models that learn from multiple modalities.
Full Lifecycle Model Development-Lead end-to-end ML workflows: data curation, preprocessing, training, evaluation, and deployment.-Collaborate with engineers to integrate ML models into production C++ pipelines and real-time systems.-Optimize inference performance and ensure robustness, reproducibility, and traceability of results.
Cross-Functional Collaboration-Work closely with product managers, clinical experts, software engineers, and hardware teams to align technical development with clinical needs.-Translate healthcare challenges into AI-driven solutions that improve surgical workflows and outcomes.-Communicate complex technical concepts clearly to both technical and non-technical stakeholders.
Research & Innovation-Stay current with the latest ML research in computer vision, medical imaging, and robotics.-Evaluate emerging approaches in unsupervised/self-supervised learning and privacy-preserving ML.-Contribute to intellectual property and scientific publications, where appropriate.
Ethical and Reliable AI-Champion the development of explainable, fair, and clinically trustworthy models.-Design data pipelines and algorithms that mitigate bias and ensure reproducibility in healthcare applications.
What We're Looking For Required Qualifications
- Masterâs or PhD in Computer Science, Electrical Engineering, Biomedical Engineering, or a related field (PhD preferred).
- 4+ years (2+ years if PhD with computer vision research) Â of experience in ML model development for computer visionâindustry or postdoctoral research.Â
- Strong Python and C++ programming skills; experience integrating models into production systems.
- Proficiency in deep learning frameworks such as PyTorch, TensorFlow, or Keras.
- Hands-on experience with medical imaging modalities (MRI, ultrasound) or a strong willingness to learn.
- Solid understanding of model evaluation, hyperparameter tuning, and dataset curation.
- Familiarity with database systems for managing large-scale medical image datasets.
- Ability to work in agile, cross-disciplinary environments and manage technical tasks in Jira or similar tools.
Preferred Qualifications
- Published work or patents in the field of AI for healthcare or medical imaging.
- Experience with video analysis, temporal models, or multi-modal data fusion.
- Strong background in image registration, tracking, or 3D model reconstruction.
- Familiarity with image processing libraries (e.g., OpenCV, ITK, SimpleITK).
- Experience deploying models on edge devices or embedded systems for real-time inference.
- Knowledge of explainability tools (e.g., Grad-CAM, SHAP) and model fairness assessment.
EQUAL EMPLOYMENT OPPORTUNITY STATEMENTPROCEPT BioRobotics is an equal opportunity employer that is committed to diversity and inclusion in the workplace. We prohibit discrimination and harassment of any kind on the basis of race, color, national origin, religion, gender, gender identity, sexual orientation, disability, genetic information, pregnancy, age, or any other protected status set forth in federal, state, or local laws. This policy applies to all employment practices within our organization. PAY RANGE TRANSPARENCYProcept is committed to fair and equitable compensation practices. The pay range(s) for this role represents a base salary range for non-commissionable roles or on-target earnings for commissionable roles. Actual compensation packages are based on several factors that are unique to each candidate, including but not limited to job-related skills, depth of experience, relevant certifications and training, and specific work location. Based on the factors above, Procept utilizes the full width of the range. The total compensation package for this position may also include eligibility for annual performance bonus, equity, and the benefits listed above. WORK ENVIRONMENTWeâll provide you training for, and ask you to maintain trained status for, and comply with, all relevant aspects of the PROCEPT BioRobotics Quality Management System to ensure product and support regulatory compliance. We would also ask you to understand and adhere to the PROCEPT BioRobotics Quality & EHS policies.
* Salary range is an estimate based on our AI, ML, Data Science Salary Index đ°
Tags: 3D Reconstruction Agile Architecture Classification Computer Science Computer Vision Data pipelines Deep Learning Engineering GANs Jira Keras Machine Learning ML models OpenCV PhD Pipelines Privacy Python PyTorch R Research ResNet Robotics TensorFlow Transformers YOLO
Perks/benefits: 401(k) matching Career development Equity / stock options Fitness / gym Flex hours Flex vacation Health care Insurance Medical leave Parental leave Salary bonus Startup environment Wellness
More jobs like this
Explore more career opportunities
Find even more open roles below ordered by popularity of job title or skills/products/technologies used.