Job Specific Information

Title:

AI/ML Engineer Medical Imaging

Location:

Pune/Gurugram

Purpose Statement

The AI/ML Engineer will lead the design and implementation of advanced imaging algorithms for a next-generation healthcare platform focused on 3D reconstruction from medical modalities such as TEE and CT. This role involves applying state-of-the-art machine learning, deep learning, and computer vision techniques to enhance medical image understanding, segmentation, and reconstruction. The candidate will collaborate with cross-functional teams including frontend, backend, QA, DevOps, and clinicians to build AI-driven imaging solutions.

Key Responsibilities

Design, develop, and optimize AI/ML algorithms for medical image analysis, segmentation, and 3D reconstruction from TEE and CT images.
Research and implement advanced deep learning architectures including CNNs, GANs, VAEs, and Diffusion Models for medical imaging tasks.
Develop robust 3D reconstruction pipelines from 2D image data and multi-view geometries, tailored to medical imaging workflows.
Perform multimodal image registration (CT-CT, CT-MRI, Fluro-Endo, 2D-3D) and develop tools for alignment, calibration, and fusion.
Enhance and denoise medical images using advanced computer vision and AI-based enhancement techniques.
Work extensively with DICOM data, integrating with PACS systems for data ingestion and retrieval.
Collaborate with teams for dataset curation, labeling, and ground truth generation.
Develop scalable training and inference pipelines on cloud platforms (AWS preferred; Azure/GCP acceptable).
Ensure reproducibility and traceability in experiments using MLOps practices (Docker, MLflow, or similar).
Collaborate with software engineers to integrate AI components into production-grade imaging applications.
Document research findings, maintain version-controlled repositories, and contribute to technical publications or IP filings.
Stay up-to-date with emerging trends in AI, computer vision, and medical imaging technologies.

Role Requirements

Bachelor's, Master's, or Ph.D. in Computer Science, Data Science, Biomedical Engineering, or related field with focus on AI, ML, or Computer Vision.
8+ years of hands-on experience in AI/ML model development with strong exposure to computer vision and imaging applications.
Expert-level proficiency in Python and deep learning frameworks such as PyTorch and TensorFlow.
Experience in 2D and 3D medical imaging (CT, MRI, Ultrasound, TEE) and DICOM data handling.
Strong understanding of 3D geometry, camera calibration, stereo vision, and multi-view reconstruction.
Experience in segmentation, registration, and object tracking within medical image contexts.
Proficiency with classical computer vision techniques (OpenCV, PCL, feature detection, structure-from-motion, SLAM, etc.).
Knowledge of generative and reconstruction models (GANs, VAEs, Diffusion Models) and fine-tuning methods for domain-specific applications.
Experience with data preprocessing, augmentation, and pipeline automation for large-scale medical datasets.
Familiarity with MLOps, containerization (Docker), and deployment workflows for cloud and edge environments.
Experience using cloud platforms (AWS, Azure, or GCP) for model training and large dataset management.
Strong software engineering fundamentals version control (Git), testing, CI/CD, and documentation practices.
Excellent analytical, communication, and collaboration skills with a strong commitment to quality and compliance in healthcare development.

Preferred / Secondary Skills

Experience in endoscopy imaging, image mosaicing, and fusion with ultrasound imaging.
Experience in 3D visualization, rendering, and medical image annotation tools.
Knowledge of reinforcement learning or self-supervised learning techniques for imaging applications.
Background in signal processing or physics-based imaging reconstruction methods.
Exposure to regulatory and quality systems in medical device software development (e.g., ISO 13485, IEC 62304).

Quality System Requirements

Demonstrates a primary commitment to patient safety and product quality by maintaining compliance to the Quality Policy and all other documented quality processes and procedures.
Establishes and promotes a work environment that supports the Quality Policy and Quality System.

H

Senior Software Developer, Embedded AI/ML Platform

Senior Software developer

UI Software Developer

Senior Software Developer

Senior Software Developer

Instrument Software Developer

iOS Software Developer