Senior Software developer

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.

• 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