Senior AI/ML Engineer (India/Remote)

About the Role

We are hiring an AI/ML Engineer based in India. You will help design, develop, optimize, and deploy a multimodal AI models for eye disease screening using image and tabular/textual data.

You will collaborate closely with AI researchers, engineers, and product teams to build and translate cutting-edge models. This is an opportunity to work at the frontier of clinical AI with purpose-driven colleagues and powerful social impact.

Job Type: Full-Time, Remote
Start Date: Immediate
Compensation: 35-50 lakh INR, commensurate with experience

Responsibilities

• Design and implement multimodal deep learning models combining image encoders and language models.

• Train, fine-tune, and optimize models using annotated eye images and structured clinical data.

• Implement instruction-tuned outputs for diagnosis, referral decisions, and patient counseling in English, Hindi, and Tamil.

• Optimize inference performance using quantization, pruning, and model distillation for deployment on smartphones or edge devices.

• Work with mobile and backend engineers to integrate models with our telemedicine app and cloud-based infrastructure.

• Contribute to model evaluation across clinical sites using real-world patient data to measure accuracy, bias, and latency.

• Support development of responsible AI pipelines: privacy, bias mitigation, versioning, and drift detection.

Qualifications

Must-Have:

• Bachelor’s or Master’s degree in Computer Science, AI, Biomedical Engineering, or related field.

• 4+ years of experience with deep learning framework (Master’s degree can substitute for experience).

• Hands-on experience training or fine-tuning transformer models (e.g., LLaMA, T5, GPT).

• Hands-on experience working with vision models (CNNs, ViTs)

• Fluency with version control (Git), collaborative workflows, and cloud-based development.

• Passion for using AI in global health or social impact domains.

Preferred:

• Experience with multimodal fusion techniques (cross-attention, late fusion, MLP).

• Experience implementing or optimizing Retrieval-Augmented Generation (RAG) pipelines for domain-specific applications (e.g., medical QA, knowledge-grounded generation).

• Experience working with multilingual NLP/NLG (especially Hindi and Tamil).

• Prior work with model optimization for edge deployment using ONNX, Core ML, TensorFlow Lite, or quantized PyTorch models, or knowledge of hybrid cloud/on-device design.

• Proven ability to mentor junior engineers and foster a culture of technical growth and collaboration.

• Strong written and verbal communication skills with cross-functional stakeholders (e.g., product, clinical, design)

Why Join Us

• Work on a mission-driven project with real clinical impact across underserved communities in India.

• Collaborate with leading experts from Johns Hopkins, Aravind Eye Hospital, Google, and other global partners.

• Flexibility to work remotely while contributing to a globally recognized AI/health project. (Note: Must be available for 2–3 regularly scheduled Zoom meetings per week and one in-person week per year.)

• Join a startup-minded team with stable long-term partnerships and funding.