Senior AI/ML Engineer

Job Description – Senior AI/ML Engineer & Python Developer

Location: Hyderabad

Experience: 4–6 years

Department: AI & Platform Engineering

Type: Full-Time | Core Product Team

1. Role Overview

We are looking for a highly skilled Senior AI/ML Engineer (with strong Python expertise) to build AI-first capabilities for our Unified Intelligent Digital Health Platform. This role blends deep AI/ML expertise with production-grade Python development. You will drive innovative GenAI, RAG-based assistants, predictive health models, and scalable backend services that power personalized digital health experiences for millions of users.

You will work across NLP, LLMs, clinical ontology models, knowledge graphs, vector search, Inference engine, ML microservices, and event-driven pipelines within a multi-tenant, secure, healthcare-grade cloud platform.

2. Key Responsibilities

A. AI/ML Engineering (Core Responsibility)

Model Development & Research

• Design, train, and optimize ML/DL models for:
• Women's health, preventive health analytics, risk prediction
• NLP (symptom extraction, clinical summarization, entity recognition)
• LLM-based conversational agents, health companion bots
• Recommendation systems (care plans, lifestyle recommendations)

Generative AI & RAG Systems

• Build domain-specific LLM applications with:
• Retrieval-Augmented Generation (RAG)
• Vector DBs (Pinecone / Qdrant / Weaviate)
• Knowledge graphs, semantic search
• Custom embedding pipelines for medical corpora

Production ML Systems

• Develop and deploy scalable ML microservices using Python & FastAPI.
• Implement MLOps (MLflow, Kubeflow, Airflow, SageMaker, ZenML).
• Model monitoring, drift detection, and real-time inference optimization.

Data Engineering Collaboration

• Partner with data engineers to build feature stores, ETL pipelines, and curated datasets.
• Ensure data quality, privacy, and compliance (HIPAA-like, PHI/PII masking).

B. Python Development (Backend & Platform Services)

Backend Engineering

• Build Python-based microservices using FastAPI, Flask, or Django.
• Implement health data workflows, rule engines, device integration adapters, and secure APIs.
• Contribute to synchronous and event-driven architectures (Kafka / AWS SNS/SQS).

System Architecture & Scaling

• Design modular backend components for a multi-tenant SaaS health platform.
• Implement authentication, RBAC, audit logs, quota management, and platform observability.

Data Handling & Storage

• Work with SQL/NoSQL (PostgreSQL, MongoDB), S3, caching layers, Redis, time-series DBs.
• Optimize performance for ingestion, transformations, and retrieval.

DevOps & Cloud (Good to Have)

• Work on containerized deployments (Docker, Kubernetes, Helm).
• Build CI/CD automation for AI and backend services.
• Deploy and optimize workloads on AWS.

3. Required Skills & Qualifications

Core Technical Skills

• Excellent Python development skills (OOP, async programming, design patterns).
• Deep understanding of ML/DL algorithms:
• Transformers, embeddings, NLP pipelines
• Time-series forecasting
• Classification, clustering, recommendation models
• Experience with frameworks:
• PyTorch, TensorFlow, scikit-learn
• HuggingFace, LangChain
• Strong background in:
• GenAI, LLM integration
• RAG systems & vector databases
• Model optimization, quantization, and inference efficiency
• Experience with REST APIs, microservices, and ML microservices.
• Strong experience with AWS (SageMaker, Lambda, S3, EC2, IAM).

Data & Platform Skills

• SQL/NoSQL, Parquet, data modeling
• Data pipelines, ETL/ELT, streaming
• API Gateway and secure API design
• Experience with healthcare/clinical datasets is a plus
• (ICD-10, FHIR, SNOMED CT, medical ontologies)

Soft Skills

• Strong analytical thinking and problem-solving
• Excellent communication — explain AI to product, clinical, and engineering teams
• Ownership mindset, startup adaptability
• Ability to work cross-functionally with design, product, and engineering

4. Preferred Experience (Bonus but Not Mandatory)

• Experience in digital health / wellness / FemTech
• Building RAG + LLM-based health assistants
• Exposure to clinical workflows and regulatory frameworks (HIPAA, GDPR)
• Experience with multi-tenant SaaS platforms
• Prior startup experience or research publications
• Contributions to patents, open-source, or internal innovation initiatives

5. Education

• B.Tech / M.Tech / MS / PhD in Computer Science, AI/ML, Data Science or related fields
• Equivalent industry experience is also considered

6. What You Will Get

• Opportunity to shape the AI backbone of a next-gen Intelligent Digital Health Platform
• Work on LLMs/SLMs, RAG, predictive health models, and real-time analytics
• High ownership, visibility, and ESOP opportunities
• Collaborative culture with a focus on research, experimentation, and product impact

Job Type: Full-time

Pay: ₹1,200, ₹2,000,000.00 per year

Benefits:

• Health insurance
• Leave encashment
• Paid time off

Work Location: In person