Python Developer - Machine Learning

Role Summary

We're looking for a hands-on Senior ML Engineer (610 years) who can:

Lead design and delivery of LLM & Document AI solutions end-to-end

Own ML services in production (APIs, pipelines, monitoring)

Mentor a team of ML engineers and work directly with clients (US & India)

This is a technical leadership role, not a pure manager role.

Key Responsibilities

Own the end-to-end lifecycle of AI features from requirement understanding

and solution design to development, deployment and monitoring.

Build LLM/RAG-based solutions using vector databases (e.g., pgvector, FAISS,

Chroma, OpenSearch) for search, Q&A, extraction and summarization over

documents.

Develop Document AI models / pipelines for OCR post-processing, entity/key

value extraction, classification and layout/table understanding.

Deploy models as scalable APIs / microservices in cloud and on-prem

environments using Python, Docker and CI/CD.

Work directly with client stakeholders to refine requirements, present solutions

and handle reviews / clarifications.

Act as technical lead on 23 AI projects, guide the team on architecture, design

and best practices.

Mentor junior ML engineers via code reviews, design discussions and technical

coaching.

Design with security, privacy and cost in mind, especially for data-sensitive

domains (BFSI, healthcare, legal, HR).

Must-Have Requirements Experience

610 years total experience, including 4+ years in ML / DL / NLP in production environments.

Proven experience taking ML solutions from PoC to production (not just notebooks).

Technical Skills

Strong in Python and core CS fundamentals (data structures, algorithms, clean code).

Hands-on with ML / NLP / LLMs classification, NER, QA, summarization, document understanding.

Experience with some of: PyTorch / TensorFlow / Hugging Face / LangChain / LlamaIndex / spaCy.

Built and deployed RESTful ML services / APIs integrated with web or mobile applications.

Worked with Docker and CI/CD pipelines; experience on AWS or GCP.

Good SQL skills and experience with at least one RDBMS (PostgreSQL/MySQL).

Understanding of MLOps in production monitoring, logging, model versioning, rollback, performance & cost optimisation.

Soft Skills

Excellent spoken and written English; able to explain complex ideas to nontechnical stakeholders.

Comfortable interacting with international clients and leading requirement /review calls.

Strong ownership mindset and ability to lead and mentor junior engineers.

Willingness to work with partial US time-zone overlap for key meetings when required.

Good-to-Have

Experience in Logistics, Legal, HR/Recruitment, Healthcare, BFSI, Education,Engineering/Plant drawings.

Exposure to Vision Transformers / advanced CV for document layout and OCR enhancement.

Experience with experiment tracking tools (MLflow, DVC, W&B, etc.).

Exposure to security & compliance (CIS, SOC2, GDPR, HIPAA, PCI-DSS, ITIL).

Experience with on-prem / VPC-isolated deployments and data-sensitive environments.

Knowledge of a systems language (Go / Rust / C++ / Scala) is a plus, not

mandatory.

Education

Bachelor's degree in Computer Science / Engineering / related

field (mandatory).

Master's degree is a plus