AI Engineer

About the Role

We are seeking a dedicated full-time AI Engineer to join our team in Pune. This role involves designing and deploying agentic AI and large language model (LLM) powered features, with Python as the primary programming language driving our AI services and automation capabilities.

Key Responsibilities

• Lead the full lifecycle development of agentic LLM features, from initial prototypes to robust production-ready Python applications.
• Develop and enhance AI agents that leverage LLMs and agentic workflows to automate processes such as reconciliation, reordering, and vendor billing.
• Create retrieval-augmented generation (RAG) and agentic functionalities that convert raw case details into reliable, actionable insights.
• Design, manage, and maintain data pipelines essential for model training and analytics.
• Assess, fine-tune, and supervise deployed models, ensuring high accuracy especially due to critical clinical and billing contexts.
• Collaborate cross-functionally to translate clinical workflows into AI-driven system features and contribute to architectural design, code reviews, and roadmap planning.
• Monitor AI systems in production, responding swiftly to real-world feedback and edge cases to improve performance.

Required Qualifications

• Minimum 2 years professional software engineering experience; at least 1 year developing ML/AI features in production environments.
• Advanced proficiency in Python programming and practical skills with ML frameworks such as PyTorch or TensorFlow.
• Hands-on experience constructing solutions with LLMs and agent frameworks (e.g., LangChain, LlamaIndex), including RAG pipelines and prompt engineering.
• Proficient in designing and running data pipelines (both batch and streaming) that feed ML models and analytics.
• Experience creating and deploying REST APIs and microservices within cloud platforms (AWS, GCP, or Azure) using Docker containers.
• Thorough understanding of ML model lifecycle management, including monitoring and iterative improvements.
• Strong communication, leadership capabilities, and problem-solving aptitude.
• Bachelor's degree in Computer Science, Engineering, or a related technical discipline.

Preferred Extras

• Background in HealthTech, MedTech, or other regulated clinical software sectors.
• Expertise in MLOps practices such as model versioning, continuous integration/deployment for ML, and feature stores.
• Experience with computer vision or optical character recognition (OCR) technologies.
• Comfort working in fast-paced startup settings characterized by ambiguity and dynamic priorities.
• Knowledge of healthcare data standards and regulations.

What We Offer

• Full-time, onsite position based out of our Pune office.
• Competitive compensation aligned with experience.
• Opportunity to take full ownership of AI products in a clinical environment, allowing rapid shipping of impactful features within a small, agile team.