About The Role

As an AI/ML Engineer at IQWorks, you will work on the AI that powers our platform. You will build, fine-tune, and deploy models for PII detection, named entity recognition, data classification, and document understanding. A key focus of this role is optimizing and deploying models that run on-premise and at the edge, enabling enterprises to process sensitive data without it ever leaving their infrastructure. Your work will directly impact how enterprises discover and protect sensitive data across structured and unstructured sources.

What You Will Do

Build, fine-tune, and adapt NLP models for PII detection, named entity recognition, and text classification
Build and maintain the data classification capabilities behind our data discovery and classification products
Design and implement model evaluation frameworks to measure precision, recall, and accuracy across data types
Optimize models for on-premise and edge deployment — small footprint, low latency, and offline-capable inference
Optimize model inference for cloud production workloads — latency, throughput,



and cost efficiency
Collaborate with product and engineering teams to integrate AI capabilities into the platform
Research and prototype current approaches for document understanding, contextual classification, and anomaly detection
Build training data pipelines and labeling workflows to continuously improve model quality

What We Are Looking For

3+ years of experience in machine learning engineering or applied AI
Strong proficiency in Python and modern ML frameworks
Hands-on experience with NLP tasks: NER, text classification, sequence labeling, or document understanding
Experience deploying models to production in both cloud and on-premise/edge environments
Solid understanding of ML fundamentals: training, evaluation, overfitting, data quality, and model lifecycle
Familiarity with vector databases, embeddings, and retrieval-augmented generation (RAG) patterns
Effective use of AI as a force mult

📌 AI/ML Engineer (Mumbai)
🏢 IQWorks
📍 Mumbai