Description

About the Role

We are hiring an AI/ML Engineer to design, develop, optimize, and deploy 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.

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.

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.