Forward Deployed ML Engineer

Important:

To successfully apply for this position, filling out this form is mandatory . Applications without a completed form will not be considered .

Position: Forward Deployed ML Engineer

Location: On-site

Type: Full Time

About the Role:

As a Forward Deployed ML Engineer, you will build and deploy cutting-edge healthcare AI solutions powered by LLMs directly alongside customers and internal stakeholders. You will work end-to-end from discovery and scoping, to rapid prototyping, to production deployment, owning outcomes in real clinical environments.

Your primary focus will be extracting structured clinical data from unstructured clinical notes and medical documents using RAG, advanced prompting, context engineering, hybrid retrieval, re-ranking, and agentic workflows. This is a high-ownership, high-velocity role where you iterate quickly, ship frequently, and continuously improve accuracy, reliability, and robustness based on real-world feedback.

This role is ideal for someone who loves being close to the problem, thrives in ambiguity, and can balance experimentation with strong engineering discipline.

What We Expect You to Bring:

1. Bachelor's or Master's degree in Computer Science or related field.
2. 2+ years of professional experience (or 1+ years with an advanced degree) in building and deploying ML/NLP systems.
3. Strong programming skills in Python, with experience building production-grade services, APIs, or pipelines.
4. Hands-on experience with LLMs (e.g., OpenAI, Anthropic, LLaMA, Mistral, etc.) and at least one LLM framework (e.g., LangChain, LlamaIndex, Haystack).
5. Strong grasp of prompt engineering and context management (system prompts, few-shot examples, retrieval configuration, memory, tool use, etc).
6. Practical experience with RAG pipelines, including embeddings, document chunking, vector databases (FAISS, Chroma, Pinecone, Pgvector), retrieval strategies, and re-ranking.
7. Understanding of agentic workflows (LLM agents that can plan, call tools/APIs, and execute multi-step tasks).
8. Familiarity with ML/AI fundamentals (transformers, fine-tuning, evaluation, error analysis).
9. Comfort working in a fast-paced, iterative environment, with strong problem-solving skills and a builder mindset.
10. Experience translating messy, real-world requirements into practical solutions.
11. Ability to own deliverables end-to-end (design, prototype, production, monitoring, iteration)

What You Will Be Doing:

1. Design and build LLM-powered healthcare applications focused on high-accuracy extraction of clinical entities, attributes, and relationships from unstructured clinical notes and documents.
2. Integrate and orchestrate open-source and proprietary LLM models into robust, production-ready systems.
3. Develop end-to-end RAG pipelines: data ingestion, preprocessing, chunking, indexing, retrieval, re-ranking, and response generation.
4. Rapidly prototype solutions in days, validate with real data and users, and evolve prototypes into production-quality code.
5. Continuously experiment with and refine prompting strategies and context engineering (system prompts, few-shot examples, retrieval configurations, document chunking strategies, hybrid retrieval techniques, smart chain-of-thought reasoning, and other advanced methods for information extraction from large document collections) to drive measurable gains in extraction accuracy and reliability.
6. Build and refine agentic frameworks using LLMs, enabling AI agents to autonomously query knowledge bases, call tools/APIs, and execute multi-step workflows.
7. Define and implement evaluation frameworks (quantitative and qualitative) for LLM-based clinical information extraction systems, including hallucination control and safety guardrails.
8. Collaborate closely with clinicians, product managers, and data teams to align solutions with real-world clinical workflows and constraints.
9. Ensure data privacy, security, and compliance when working with sensitive healthcare data.
10. Stay up to date with the latest LLM, RAG, and agentic system research, and translate promising ideas into practical prototypes and product features.

Why Join?

1. We are revolutionizing a unique industry that has the potential to impact and benefit patients from all over the world - you can create impact at scale.
2. We have access to the best computing resources available, including the H100 and A100, among others.
3. We have had company-sponsored workations in Bali, Sri Lanka, and Manali and take pride in our hard-working yet super fun culture.
4. We are working on a few of the most challenging problems in a highly regulated industry, which provides you with an opportunity to solve some of the most interesting things.
5. You will get a chance to work with experts from multiple industries, the best in the industry compensation, and to continue building your own (and, of course, new) projects.

Perks & Benefits:

• Unlimited Leave Policy take time off when you need it. We believe in trust, not tracking.
• Lunch Provided at the Office one less daily decision, one happier employee.
• Flexible Working Hours we care about output, not clock-ins.
• Health Insurance comprehensive coverage for you and your family.
• Zomato Meal Benefit breakfast and dinner can be ordered when you come in early or leave late, because effort deserves fuel.