Senior AIML Engineer @ Madurai at Techmango Technology Services Private Limited · Madurai · 5 - 10 years · ₹15L - ₹20L / yr · Profitable · Posted 9 Jul 2025

Senior AIML Engineer @ Madurai
Role : AIML Engineer
Location : Madurai
Experience : 5 to 10 Yrs
Mandatory Skills : AIML, Python, SQL, ML Models, PyTorch, Pandas, Docker, AWS
Language: Python
DBs : SQL
Core Libraries:
Time Series & Forecasting: pmdarima, statsmodels, Prophet, GluonTS, NeuralProphet
SOTA ML : ML Models, Boosting & Ensemble models etc.
Explainability : Shap / Lime
Required skills:
- Deep Learning: PyTorch, PyTorch Forecasting,
- Data Processing: Pandas, NumPy, Polars (optional), PySpark
- Hyperparameter Tuning: Optuna, Amazon SageMaker Automatic Model Tuning
- Deployment & MLOps: Batch & Realtime with API endpoints, MLFlow
- Serving: TorchServe, Sagemaker endpoints / batch
- Containerization: Docker
- Orchestration & Pipelines: AWS Step Functions, AWS SageMaker Pipelines
AWS Services:
- SageMaker (Training, Inference, Tuning)
- S3 (Data Storage)
- CloudWatch (Monitoring)
- Lambda (Trigger-based Inference)
- ECR, ECS or Fargate (Container Hosting)

About Techmango Technology Services Private Limited
About
Founded in 2014, Techmango Technology Services (TTS) is a dynamic IT services and AI consulting company that has steadily expanded into a robust organization with a team of over 250 highly skilled professionals. A NASSCOM Partnered and CMMi Level 3 initiated Madurai-based IT company, with a difference. As a client centered and quality conscious IT company, we offer a wide spectrum of technology services and solutions to help our clients meet their business needs on time and within cost-effective parameters.
Techmango strives to deliver top-tier GenAI, Data Engineering solutions, App Modernization and Custom Software Development to keep you ahead of the competition, continually enhancing your IT-driven business solutions.
We leverage the latest software development platforms, tools and advanced project management techniques to deliver industry-leading solutions that set you apart from the competition. Our comprehensive service portfolio includes Application Services, Infrastructure Services, and Professional Services, all delivered through our flexible Global Delivery Model.
Similar jobs (10)
Job Title: Senior AI/ML Engineer
Company: Timble Technologies Pvt. Ltd
Location: Gurugram (Hybrid)
Experience: 2 TO 5 Years
About Us
Timble Glance is a high-growth AI RegTech and B2B SaaS company catering to top-tier BFSI and enterprise clients. We build cutting-edge systems powering 30+ high-scale APIs for digital identity verification, fraud detection, document intelligence, and compliance automation.
Role Overview
We are looking for a hands-on Senior AI/ML Engineer to design, develop, and productionize high-throughput AI/ML and Generative AI systems. You will own the full lifecycle—from problem formulation and data pipelines to deep learning architectures, RAG systems, LLMOps, and model governance—delivering sub-second latency and high reliability across our enterprise products.
Key Responsibilities
· Model Architecture & Deployment: Design, train, and deploy production-scale ML/Deep Learning and GenAI systems (computer vision, document intelligence, OCR, NLP, fraud risk classification, and LLM applications).
· GenAI & LLM Solutions: Develop robust LLM workflows including prompt engineering, fine-tuning, RAG pipelines, semantic search, vector indexing (Pinecone/Milvus/Chroma), and safety guardrails.
· Pipelines & Engineering: Build performant feature extraction and data pipelines; write modular, vectorized, production-grade Python (NumPy, Pandas) and advanced SQL.
· MLOps & Monitoring: Establish end-to-end MLOps/LLMOps standards—model registries, CI/CD, experiment tracking, drift detection, A/B testing, latency optimization, and cost governance.
· Responsible AI & Security: Ensure model decisions comply with enterprise data security, privacy standards, and auditability required by the BFSI sector.
· Collaboration & Ownership: Translate complex business requirements into technical roadmaps, conduct rigorous code reviews, and mentor junior engineers.
Required Qualifications & Skills
· Education: B.Tech / M.Tech in Computer Science, AI/ML, Mathematics, or a related field—Tier-1 institutes (IIT, IIIT, NIT) strongly preferred.
· Experience: 2+ years of hands-on experience developing, deploying, and maintaining ML/Deep Learning or GenAI models in production environments.
· GenAI & NLP Stack: Hands-on experience with LLMs, embeddings, RAG architectures, and frameworks such as LangChain, LlamaIndex, or Hugging Face.
· Deep Learning Frameworks: Strong proficiency in PyTorch or TensorFlow, with deep knowledge of transformer architectures and modern NLP/CV models.
· Software & Data Engineering: Expert-level Python skills (pytest, Git, OOP, asynchronous programming), solid SQL proficiency, and familiarity with data workflows.
· Deployment & Cloud: Practical exposure to cloud platforms (AWS/GCP), containerization (Docker), API frameworks (FastAPI/Flask), and basic orchestration (Kubernetes).
Preferred Qualifications
· Prior domain experience in Fintech, RegTech, Identity Verification (KYC/AML), Fraud Intelligence, or B2B SaaS.
· Experience optimizing models for low latency and inference cost (e.g., ONNX, TensorRT, model quantization).
· Familiarity with workflow orchestrators such as Airflow, Prefect, or Kubeflow.
Strong Senior AI Engineer Profile with production-grade LLM and Python expertise
2
Mandatory (Experience 1): Must have at least 3+ years of professional AI/ML engineering experience with demonstrated track record of delivering production-grade AI systems in real-world environments
3
Mandatory (Tech skill 1): Must have strong programming skills in Python and SQL, with hands-on ML/data libraries (scikit-learn, pandas, numpy) and deep-learning frameworks (PyTorch or TensorFlow), plus REST API design
4
Mandatory (Experience 2): Must have hands-on experience building and deploying production-grade ML/LLMs including RAG pipelines, document parsing, information extraction, and text processing on large-scale unstructured data (preprocessing, chunking, embeddings, feature engineering)
5
Mandatory (Tech skill 2): Must have strong NLP / extraction-focused ML depth — transformers, embeddings, vector databases, RAG, LLM integrations, and agentic workflows.
6
Mandatory (Tech skill 3): Must have hands-on experience with AWS (SageMaker, Bedrock, EC2, Lambda) and Azure (AI Foundry, Azure OpenAI) for model training, fine-tuning, deployment, and inference at scale
7
Mandatory (Tech skill 4): Must have experience with multi-agentic frameworks / orchestration tools (Claude Code, LangGraph, LangChain, CrewAI)
8
Mandatory (Tech skill 5): Must have hands-on experience with MLOps ecosystem including experiment tracking (MLflow, Weights & Biases), model versioning/registry, automated retraining, and CI/CD (GitHub Actions, Azure DevOps, Docker, Kubernetes)
9
Mandatory (Tech skill 6): Must have experience with evaluation frameworks (precision, recall, F1, field-level accuracy) and iterative model improvement, plus AI observability (Prometheus, Grafana, SLOs).
10
Mandatory (Team Leadership): Must have experience leading projects or teams, managing technical deliverables, and mentoring, with strong problem-solving in ambiguous challenges
11
Mandatory (Education) - B.Tech or B.E. If graduated in BCA or BCS, they should have a Maters or Post graduation
12
Mandatory (Company): B2B Tech/Consulting/B2B Software
Strong Senior AI Engineer Profile with production-grade LLM and Python expertise
2
Mandatory (Experience 1): Must have at least 3+ years of professional AI/ML engineering experience with demonstrated track record of delivering production-grade AI systems in real-world environments
3
Mandatory (Tech skill 1): Must have strong programming skills in Python and SQL, with hands-on ML/data libraries (scikit-learn, pandas, numpy) and deep-learning frameworks (PyTorch or TensorFlow), plus REST API design
4
Mandatory (Experience 2): Must have hands-on experience building and deploying production-grade ML/LLMs including RAG pipelines, document parsing, information extraction, and text processing on large-scale unstructured data (preprocessing, chunking, embeddings, feature engineering)
5
Mandatory (Tech skill 2): Must have strong NLP / extraction-focused ML depth — transformers, embeddings, vector databases, RAG, LLM integrations, and agentic workflows.
6
Mandatory (Tech skill 3): Must have hands-on experience with AWS (SageMaker, Bedrock, EC2, Lambda) and Azure (AI Foundry, Azure OpenAI) for model training, fine-tuning, deployment, and inference at scale
7
Mandatory (Tech skill 4): Must have experience with multi-agentic frameworks / orchestration tools (Claude Code, LangGraph, LangChain, CrewAI)
8
Mandatory (Tech skill 5): Must have hands-on experience with MLOps ecosystem including experiment tracking (MLflow, Weights & Biases), model versioning/registry, automated retraining, and CI/CD (GitHub Actions, Azure DevOps, Docker, Kubernetes)
9
Mandatory (Tech skill 6): Must have experience with evaluation frameworks (precision, recall, F1, field-level accuracy) and iterative model improvement, plus AI observability (Prometheus, Grafana, SLOs).
10
Mandatory (Team Leadership): Must have experience leading projects or teams, managing technical deliverables, and mentoring, with strong problem-solving in ambiguous challenges
11
Mandatory (Education) - B.Tech or B.E. If graduated in BCA or BCS, they should have a Maters or Post graduation
12
Mandatory (Company): B2B Tech/Consulting/B2B Software
Kody Technolab Limited
AI/ML Lead (7+ Years Experience)
Location: Ahmedabad / Gandhinagar
Experience: 7+ Years
Employment Type: Full-Time
Kody Technolab Limited is seeking an experienced AI/ML Engineer to design, develop, and deploy
cutting-edge Artificial Intelligence and Machine Learning solutions. The ideal candidate will have
strong expertise in Machine Learning, Deep Learning, Generative AI, LLMs, MLOps, and
cloud-based AI deployments.
Key Responsibilities
• Design, develop, and deploy Machine Learning and Deep Learning models for classification,
regression, recommendation systems, NLP, Computer Vision, and Generative AI applications.
• Build and maintain end-to-end ML pipelines including data preprocessing, feature
engineering, model training, validation, evaluation, and deployment.
• Develop AI solutions using PyTorch, TensorFlow, Scikit-learn, Hugging Face, and related
frameworks.
• Work with Large Language Models (LLMs) and foundation models such as GPT, BERT, Llama,
Claude, and Stable Diffusion.
• Collaborate with product, engineering, and business teams to translate requirements into
scalable AI solutions.
• Optimize model performance, scalability, and reliability for production environments.
• Implement MLOps best practices using tools such as MLflow, Docker, Kubernetes, and Kubeflow.
• Stay updated with emerging trends and research in AI, ML, Deep Learning, and Generative AI.
Required Qualifications
• Bachelor’s or Master’s degree in Computer Science, Data Science, Artificial Intelligence,
Mathematics, or a related field.
• 7+ years of hands-on experience in AI/ML product development.
• Strong proficiency in Python and ML frameworks including Scikit-learn, TensorFlow, PyTorch, and
Hugging Face.
• Experience with Generative AI, LLMs, GANs, VAEs, diffusion models, and prompt engineering.
• Strong understanding of the ML lifecycle including model training, tuning, deployment,
monitoring, and optimization.
• Experience with MLOps tools such as MLflow, Docker, Kubeflow, and CI/CD pipelines.
• Experience with AWS, Azure, or GCP cloud platforms.
• Strong problem-solving and analytical skills.Preferred Skills
• Fine-tuning and deployment of Large Language Models.
• Experience with RAG (Retrieval Augmented Generation) architectures.
• Contributions to open-source AI projects or research publications.
• Knowledge of model interpretability, data annotation, and feature engineering.
• C++ experience for high-performance AI applications.
Why Join Kody Technolab Limited?
Opportunity to work on innovative AI products, Generative AI solutions, robotics integrations,
and enterprise-scale applications while collaborating with a highly skilled technology team.
Visit the Website to know more about us.
Company Website - Kody Technolab | Deep Tech Company in Robotics & AI Solution
Kody Robots | Robotics Company in India for Autonomous Robots
Kody Technolab Limited is seeking an experienced AI/ML Engineer to design, develop, and deploy cutting-edge Artificial Intelligence and Machine Learning solutions. The ideal candidate will have
strong expertise in Machine Learning, Deep Learning, Generative AI, LLMs, MLOps, and cloud-based AI deployments.
Key Responsibilities
• Design, develop, and deploy Machine Learning and Deep Learning models for classification, regression, recommendation systems, NLP, Computer Vision, and Generative AI applications.
• Build and maintain end-to-end ML pipelines including data preprocessing, feature engineering, model training, validation, evaluation, and deployment.
• Develop AI solutions using PyTorch, TensorFlow, Scikit-learn, Hugging Face, and related frameworks.
• Work with Large Language Models (LLMs) and foundation models such as GPT, BERT, Llama, Claude, and Stable Diffusion.
• Collaborate with product, engineering, and business teams to translate requirements into scalable AI solutions.
• Optimize model performance, scalability, and reliability for production environments.
• Implement MLOps best practices using tools such as MLflow, Docker, Kubernetes, and Kubeflow.
• Stay updated with emerging trends and research in AI, ML, Deep Learning, and Generative AI.
Required Qualifications
• Bachelor’s or Master’s degree in Computer Science, Data Science, Artificial Intelligence, Mathematics, or a related field.
• 7+ years of hands-on experience in AI/ML product development.
• Strong proficiency in Python and ML frameworks including Scikit-learn, TensorFlow, PyTorch, and Hugging Face.
• Experience with Generative AI, LLMs, GANs, VAEs, diffusion models, and prompt engineering.
• Strong understanding of the ML lifecycle including model training, tuning, deployment, monitoring, and optimization.
• Experience with MLOps tools such as MLflow, Docker, Kubeflow, and CI/CD pipelines.
• Experience with AWS, Azure, or GCP cloud platforms.
• Strong problem-solving and analytical skills.
Preferred Skills
• Fine-tuning and deployment of Large Language Models.
• Experience with RAG (Retrieval Augmented Generation) architectures.
• Contributions to open-source AI projects or research publications.
• Knowledge of model interpretability, data annotation, and feature engineering.
• C++ experience for high-performance AI applications.
Why Join Kody Technolab Limited?
Opportunity to work on innovative AI products, Generative AI solutions, robotics integrations,
and enterprise-scale applications while collaborating with a highly skilled technology team.
Visit the Website to know more about us.
Company Website - Kody Technolab | Deep Tech Company in Robotics & AI Solution
Kody Robots | Robotics Company in India for Autonomous Robots
ABOUT
The Persona Labs is building a new kind of social platform focused on something most social products do not explicitly optimize for: helping people become real friends.
We want to help people discover interesting people around them, find meaningful common ground, start low-pressure interactions, continue promising conversations, create shared experiences, and ultimately build real-life friendships.
DISCOVER → CURIOSITY → COMPATIBILITY → INTERACTION → UNDERSTAND → IRL EXPERIENCE → FRIENDSHIP
THE AI LAYER - COMPANION INTELLIGENCE
Alongside the platform, we are building a proactive personal AI companion that learns about the user and helps them navigate this journey through personalized recommendations, suggestions, reminders, conversations, and experiences.
THE OPPORTUNITY
We are looking for a Founding ML Engineer to build the intelligence layer of the platform from the ground up. This is a 0→1 Applied AI / ML role where you will work directly with the founder and Product Engineer to turn ambiguous problems around users, relationships, recommendations and personal intelligence into working systems.
You will be expected to:
Understand the problem → identify the signals → design the intelligence system → prototype → evaluate → deploy → learn → improve.
WHAT YOU WILL BUILD & OWN
USER INTELLIGENCE
User representations, behavioural models, interests, preferences, contextual signals, and evolving understanding of the user. MEMORY Short- and long-term memory, episodic/preference/relationship memory, retrieval, relevance and updating.
RECOMMENDATION & MATCHING
People discovery, compatibility, activity/experience recommendations, and personalized ranking.
INTENT & INTEREST
Infer what the user is trying to do and learn what they care about from behaviour, not only declared interests.
RANKING
Decide what should appear first across potentially thousands of relevant people, activities or experiences.
CONTENT INTELLIGENCE
Classification, toxicity, spam, policy signals, quality, relevance, and semantic understanding.
RELATIONSHIP INTELLIGENCE
Reciprocity, interaction health, shared interests, progression, declining engagement and shared activity.
NEXT-BEST-ACTION
Determine the most useful action now: show a person, suggest a question, recommend an activity, reconnect, or do nothing.
TRUST / SAFETY INTELLIGENCE
Fake-account signals, spam, abuse, behavioural anomalies, risky interactions and moderation assistance.
COMPANION INTELLIGENCE
Use signals and outputs to help the companion decide what to say, suggest, recommend or not do.
WHAT YOUR DAY-TO-DAY LOOKS LIKE
• Translate ambiguous product problems into ML/AI system designs.
• Build models and intelligence pipelines using behavioural, relational and contextual signals.
• Develop recommendation, matching and personalization systems.
• Design memory and retrieval systems that help the companion understand the user over time.
• Build and evaluate LLM-powered and agentic workflows.
• Decide when to use traditional ML, rules, retrieval, ranking or LLMs.
• Prototype quickly, test assumptions and iterate based on real user behaviour.
• Work closely with the founder and Product Engineer to turn intelligence into product experiences.
• Design APIs and production systems that bring ML/AI capabilities into the application.
• Build evaluation, monitoring and feedback loops so the intelligence improves over time.
WHO SHOULD APPLY
• Experience: 0–4 years’ experience, including exceptional fresh graduates. Strong 1–3 year engineers and experienced 3–4 year product builders are welcome.
• Strong foundations in ML, Python, statistics and software engineering.
• Evidence of Building: Experience with AI/ML projects, recommendation systems, LLM applications or personalization is highly valued.
• Strong evidence of building: Shipped projects, research, hackathons, internships, open source or startup work.
WHAT WE LOOK FOR
MACHINE LEARNING DEPTH
Can you understand the modelling problem underneath the application?
RECOMMENDATION & PERSONALIZATION
Can you reason about relevance, ranking, cold start and behavioural signals?
AI ENGINEERING
Can you turn LLMs and agents into reliable product capabilities rather than simple API wrappers?
USER INTELLIGENCE
Can you design systems that gradually understand a person from sparse and changing signals?
SYSTEMS THINKING
Can you move from a model to a production system with APIs, data, latency, cost and monitoring?
EVALUATION MINDSET
Can you determine whether the intelligence actually helped the user?
PRODUCT JUDGMENT
Can you decide what the system should do when there is no predefined answer?
SPEED OF EXECUTION
Can you move from idea → prototype → evaluation → production quickly and responsibly?
BUILD WITH US
You will join at a stage where many of the answers do not exist yet. You will not simply implement a model someone else selected; you will help decide how the product learns to understand people.
CAREERS:
Apply with your resume, GitHub, portfolio or shipped work.
https://forms.gle/12YpUSBY2Sqs5xjp8
www.thepersonalabs.com
Role Summary
We are seeking an Advanced Machine Learning Engineer to design, build, train, deploy, and operate scalable machine learning systems. This hands-on ML engineering role focuses on implementing models and production systems—not solely conducting data analysis or research. It requires strong Python and deep-learning expertise, experience with frameworks such as PyTorch or TensorFlow, and the ability to own the end-to-end model lifecycle, from behavioral/event data preparation and feature engineering through model development, large-scale training, deployment, inference optimization, and monitoring. The engineer will work with Snowflake, Databricks, Spark, and GPU-based training infrastructure to deliver reliable, efficient, and maintainable ML capabilities.
Must have - Buulding and operated production ML systems using Python + PyTorch /TensorFlow + Snowflake/Databricks + MLOps.
Key Responsibilities
Machine Learning Engineering
- Implement and optimize machine learning algorithms and models for business use cases, including classification, regression, clustering, and recommendation systems as applicable.
- Translate model requirements into robust, reusable, and testable Python components.
- Build model training, validation, scoring, and inference workflows that can be integrated into production applications.
- Evaluate model performance against agreed technical and business metrics and implement improvements to accuracy, latency, robustness, and resource efficiency.
- Apply appropriate validation, error analysis, and model testing practices to ensure reliable behavior.
Train models efficiently on GPU infrastructure using techniques such as mixed-precision training, gradient accumulation, checkpointing, memory optimization, and distributed training strategies where appropriate.
Use PyTorch, TensorFlow,Pyspark or equivalent frameworks to implement experiments and production-ready training code; apply rigorous evaluation, error analysis, and reproducibility practices.
Build model-training workflows covering event/token sequence preparation, tokenization, temporal and positional representations, self-supervised pretraining, and downstream fine-tuning when appropriate to the use case.
Design, implement, train, and optimize deep-learning models; develop custom architectures where required, including transformer- and attention-based models for sequential or behavioral data.
Deep Learning & Model Architecture
Data Platforms & ML Pipelines
- Use Snowflake and SQL to access, query, and prepare enterprise data required by machine learning applications.
- Use Databricks for scalable data processing, feature preparation, model development, and ML pipeline execution.
- Develop and maintain reliable pipelines for data preparation, feature generation, training, batch scoring, and inference.
- Use PySpark and distributed processing techniques where appropriate for large-scale workloads.
- Work with data engineering teams to integrate upstream data sources and ensure dependable data contracts, quality checks, and pipeline operation.
Productionization & MLOps
- Productionize ML solutions and integrate them with services, APIs, data products, or downstream business workflows.
- Implement model and code versioning, experiment tracking, reproducible builds, automated testing, and release practices.
- Monitor model outputs and pipeline health; troubleshoot failures, data changes, drift, and performance degradation.
- Improve operational reliability, scalability, observability, and maintainability of ML systems.
- Collaborate with platform, cloud, and DevOps teams on deployment automation and production support.
Build observable production ML systems with appropriate logging, alerting, performance tracking, and operational safeguards.
Optimize inference latency, throughput, memory use, and serving cost through model architecture choices, batching, and serving-side optimizations.
Software Engineering & Technical Ownership
- Develop clean, modular, documented, and maintainable code following engineering standards and design principles.
- Contribute to technical design, code reviews, testing strategies, and architectural decisions for ML systems.
- Own assigned ML engineering deliverables from technical design through implementation, release, and ongoing support.
- Identify performance bottlenecks and improve runtime, resource utilization, and processing efficiency.
- Collaborate with cross-functional engineering and product teams to deliver dependable ML capabilities.
Required Skills & Qualifications
- Strong proficiency in Python and software engineering practices, including object-oriented design, modular development, testing, debugging, and code quality.
- Strong ML foundation and mathematical understanding
- Strong understanding of the end-to-end ML lifecycle, including model architecture, training, evaluation, optimization, inference, production integration, and monitoring.
- Strong SQL skills and experience working with structured data and data pipelines.
- Hands-on experience with Snowflake for querying and preparing data for ML workflows.
- Hands-on experience with Databricks for data processing and/or ML pipeline development.
- Strong hands-on experience with PyTorch, TensorFlow, or a similar deep-learning framework; experience with scikit-learn is also useful where appropriate.
- Understanding of production ML concepts, including reproducibility, model versioning, deployment, monitoring, and troubleshooting.
- Ability to design reliable, scalable, and maintainable software components and pipelines.
Familiarity with GPU training concepts and distributed strategies such as Distributed Data Parallel (DDP), Fully Sharded Data Parallel (FSDP), or equivalents is an advantage for large-model workloads.
Understanding of model optimization, inference efficiency, and integration with production services or applications.
Experience developing and deploying ML solutions at scale, including end-to-end model lifecycle ownership.
Preferred Skills
- PySpark, Apache Spark, and distributed processing for large-scale behavioral or event data.
- MLflow or similar tooling for experiment tracking, model registry, reproducible training, and model lifecycle management.
- Experience integrating Snowflake and Databricks within enterprise ML workflows.
- Cloud experience with AWS, Microsoft Azure, or Google Cloud.
- REST APIs, containers, CI/CD, automated testing, logging, and observability.
- Experience with fraud detection, anomaly detection, behavioral modeling, identity or account-abuse signals, recommendation systems, or other large-scale ML applications.
- Exposure to Generative AI, LLM integration, embeddings, or RAG as an additional advantage.
Education & Experience
- Bachelor’s or Master’s degree in Computer Science, Software Engineering, Computer Engineering, or a related field, or equivalent practical experience.
- At least 5 years of professional experience building, deploying, and supporting machine learning solutions at scale, or equivalent experience demonstrating the required engineering depth.
- Demonstrated ability to take ownership of engineering deliverables and work through implementation, deployment, and operational issues.

LLM/AI Model Engineer – Training, Evaluation & Benchmarking
Location: New Delhi, India (On-site)
Employment Type: Full-time
Experience: 3–5 years
Industry: Enterprise AI / Generative AI / Machine Learning / NLP
About QubeLabs
QubeLabs is building the next generation of Enterprise AI Systems that transform workforce operations and intelligence for the financial services industry.
Our platform combines Conversational AI, Agentic AI, workflow automation, proprietary language models and enterprise intelligence. Our flagship product, QubeLabs Workmate, serves banks, NBFCs, MFIs, wealth management firms, insurance companies and fintechs across India and Europe.
At the core of our AI platform is the Vectro Series, our proprietary family of language models designed for enterprise and financial-services use cases.
We are looking for an LLM/AI Model Engineer to build the training pipelines, datasets and evaluation infrastructure required to continuously improve the Vectro Series.
Role Overview
You will work closely with the Head of AI, Research Partners and engineering teams to implement and operationalize LLM training approaches and build a reliable model improvement lifecycle:
Data Preparation → Training → Evaluation → Benchmarking → Error Analysis → Model Improvement
The role combines hands-on LLM fine-tuning, dataset engineering, tokenization, evaluation framework development and model quality management.
Key Responsibilities
1. LLM Training & Fine-tuning
Build and maintain pipelines for training and fine-tuning open-source foundation models for the Vectro Series.
Implement supervised fine-tuning (SFT), parameter-efficient fine-tuning (PEFT), LoRA/QLoRA and domain adaptation techniques; support continued pre-training where required.
Manage training configurations, checkpoints, model versions and experiment tracking.
Optimize training workflows for computational efficiency, reproducibility and reliability.
Translate model development approaches defined with the Head of AI and Research Partners into practical, scalable training pipelines.
2. Training Data & Tokenization
Build data pipelines to collect, clean, filter, normalize, deduplicate and validate training data.
Develop instruction-tuning, SFT and domain-specific datasets for financial-services and enterprise use cases.
Implement tokenization workflows, tokenizer configuration, vocabulary management, sequence packing, truncation and padding.
Address data representation challenges involving financial terminology, numerical values and multilingual content.
Maintain dataset versioning, lineage and quality controls, incorporating identified data gaps and relevant model failures into future training data.
3. Golden Dataset & Model Evaluation
Build and maintain QubeLabs' Golden Dataset to evaluate key model capabilities and target use cases.
Develop automated evaluation pipelines and reusable evaluation harnesses, combining automated metrics with human assessment where appropriate.
Evaluate accuracy, reasoning, factuality, instruction following, safety, multilingual performance and financial-domain capabilities.
Establish consistent evaluation methodology and regression tests to measure changes across model versions.
4. Benchmarking & Error Analysis
Evaluate Vectro against relevant public benchmarks and proprietary QubeLabs BFSI/enterprise benchmarks.
Compare results with relevant open-source and commercial models using consistent evaluation conditions.
Analyze failures, identify root causes across data, training and model behavior, and translate findings into actionable improvements.
Maintain reproducible benchmark results and reports to track model strengths, limitations and progress.
5. Model Quality & Release
Define model release quality gates and ensure each major release meets agreed evaluation and benchmark criteria.
Support red-teaming, robustness and safety testing.
Coordinate with the Head of AI, Research Partners and engineering teams to ensure validated model improvements are suitable for integration into production systems.
Experience
3–5 years in Machine Learning, Deep Learning, NLP, Generative AI or related fields.
Hands-on experience training or fine-tuning LLMs using open-source foundation models.
Practical experience building training pipelines, preparing datasets and evaluating model performance.
Experience with GPU-based training; distributed training is an advantage.
Financial-services, multilingual AI or enterprise AI experience is preferred.
Required Skills
LLM & Machine Learning: Large Language Models, Transformer architectures, Generative AI, NLP, Deep Learning, SFT, PEFT, LoRA/QLoRA, fine-tuning and continued pre-training.
Frameworks & Infrastructure: Python, PyTorch, Hugging Face Transformers, Hugging Face Tokenizers, GPU computing, experiment tracking and ML pipelines.
Data & Tokenization: Tokenization, tokenizer configuration, text preprocessing, sequence packing, dataset construction, data quality, versioning and lineage.
Evaluation & Benchmarking: Golden datasets, LLM evaluation, evaluation harnesses, automated and human evaluation, LLM-as-a-Judge, public and domain-specific benchmarks, regression testing and error analysis.
The role requires the ability to build both the model training pipeline and the measurement system that determines whether the model has actually improved.
What Success Looks Like
Reliable and reproducible training and fine-tuning pipelines for the Vectro Series.
High-quality training datasets, tokenization workflows and a comprehensive Golden Dataset.
Automated evaluation and benchmarking infrastructure with measurable model quality gates.
Clear, reproducible evidence of model performance against public, proprietary BFSI and relevant competing-model benchmarks.
A continuous improvement loop that converts evaluation findings into measurable gains across successive Vectro releases.
Key Performance Indicators
Training pipeline reliability and experiment throughput.
Training data quality and coverage.
Golden Dataset and evaluation coverage.
Benchmark reproducibility and model performance improvement.
Accuracy of error diagnosis and effectiveness of regression detection.
Time required to complete training, evaluation and benchmarking cycles.
Compliance with model release quality criteria.
Why Join QubeLabs?
Build the proprietary Vectro Series of enterprise and financial-services language models.
Develop model training, evaluation and benchmarking infrastructure from the ground up.
Work closely with the Head of AI and Research Partners.
Solve real-world AI challenges across financial services.
Develop proprietary datasets, benchmarks and model improvement systems.
Contribute directly to production AI systems serving customers across India and Europe.
Preferred Candidate Profile
B.Tech, M.Tech, MS or PhD in Computer Science, AI, ML, Mathematics or a related field.
3–5 years of relevant experience with strong practical skills in Python, PyTorch, LLM fine-tuning and training data pipelines.
Working knowledge of tokenization, evaluation datasets, benchmarking and regression testing.
Experience with open-source models such as Llama, Qwen, Mistral or similar is preferred.
Ability to independently build reliable model training and evaluation systems from scratch.
QubeLabs is an equal opportunity employer and values diversity and inclusion. We encourage applications from qualified candidates regardless of background or identity.
Job Title: AI Developer
Location: Bangalore / Hyderabad
Experience: 5–7 Years
Notice Period: Immediate to 15 days Preferred
About the Role
We are looking for a skilled AI Developer to design, develop, and deploy Artificial Intelligence and Machine Learning solutions that solve real-world business problems. You will work closely with data scientists, engineers, and business stakeholders to build scalable AI-powered applications and pipelines.
Key Responsibilities
- Design and develop Machine Learning models, Deep Learning architectures, and NLP solutions for production use.
- Build and maintain end-to-end ML pipelines, from data ingestion and preprocessing to model training, evaluation, and deployment.
- Integrate AI/ML models into existing applications and APIs.
- Fine-tune and evaluate Large Language Models (LLMs) for business-specific use cases.
- Collaborate with data engineers to ensure high-quality training data.
- Monitor model performance in production and implement retraining strategies.
- Research and evaluate emerging AI frameworks, tools, and technologies.
- Document models, experiments, and technical decisions.
Required Skills & Experience
- 3+ years of experience in AI/ML development.
- Strong proficiency in Python.
- Hands-on experience with TensorFlow, PyTorch, or Scikit-learn.
- Experience with NLP frameworks such as Hugging Face, spaCy, and LangChain.
- Familiarity with LLMs and Prompt Engineering (OpenAI, Anthropic, Gemini, etc.).
- Experience deploying models using FastAPI or Flask (REST APIs).
- Strong understanding of Data Structures, Algorithms, and Statistics.
- Experience with AWS, Azure, or GCP and MLOps tools.
- Experience with Vector Databases such as Pinecone, Weaviate, or FAISS is an added advantage.
About NonStop io Technologies
NonStop io Technologies is a value-driven company with a strong focus on process-oriented software engineering. We specialize in Product Development and have a decade's worth of experience in building web and mobile applications across various domains. NonStop io Technologies follows core principles that guide its operations and believes in staying invested in a product's vision for the long term. We are a small but proud group of individuals who believe in the 'givers gain' philosophy and strive to provide value in order to seek value. We are committed to and specialize in building cutting-edge technology products and serving as trusted technology partners for startups and enterprises. We pride ourselves on fostering innovation, learning, and community engagement. Join us to work on impactful projects in a collaborative and vibrant environment.
Brief Description:
We're seeking an AI/ML Engineer to join our team. As AI/ML Engineer, you will be responsible for designing, developing, and implementing artificial intelligence (AI) and machine learning (ML) solutions to solve real-world business problems. You will work closely with engineering teams, including software engineers, domain experts, and product managers, to deploy and integrate Applied AI/ML solutions into the products that are being built at NonStop io. Your role will involve researching cutting-edge algorithms and data processing techniques, and implementing scalable solutions to drive innovation and improve the overall user experience.
Responsibilities
● Applied AI/ML engineering; Building engineering solutions on top of the AI/ML tooling available in the industry today. Eg: Engineering APIs around OpenAI
● AI/ML Model Development: Design, develop, and implement machine learning models and algorithms that address specific business challenges, such as natural language processing, computer vision, recommendation systems, anomaly detection, etc.
● Data Preprocessing and Feature Engineering: Cleanse, preprocess, and transform raw data into suitable formats for training and testing AI/ML models. Perform feature engineering to extract relevant features from the data
● Model Training and Evaluation: Train and validate AI/ML models using diverse datasets to achieve optimal performance. Employ appropriate evaluation metrics to assess model accuracy, precision, recall, and other relevant metrics
● Data Visualization: Create clear and insightful data visualizations to aid in understanding data patterns, model behaviour, and performance metrics
● Deployment and Integration: Collaborate with software engineers and DevOps teams to deploy AI/ML models into production environments and integrate them into various applications and systems
● Data Security and Privacy: Ensure compliance with data privacy regulations and implement security measures to protect sensitive information used in AI/ML processes
● Continuous Learning: Stay updated with the latest advancements in AI/ML research, tools, and technologies, and apply them to improve existing models and develop novel solutions
● Documentation: Maintain detailed documentation of the AI/ML development process, including code, models, algorithms, and methodologies for easy understanding and future reference.
Qualifications & Skills
● Bachelor's, Master's, or PhD in Computer Science, Data Science, Machine Learning, or a related field. Advanced degrees or certifications in AI/ML are a plus
● Proven experience as an AI/ML Engineer, Data Scientist, or related role, ideally with a strong portfolio of AI/ML projects
● Proficiency in programming languages commonly used for AI/ML. Preferably Python
● Familiarity with popular AI/ML libraries and frameworks, such as TensorFlow, PyTorch, scikit-learn, etc.
● Familiarity with popular AI/ML Models such as GPT3, GPT4, Llama2, BERT etc.
● Strong understanding of machine learning algorithms, statistics, and data structures
● Experience with data preprocessing, data wrangling, and feature engineering
● Knowledge of deep learning architectures, neural networks, and transfer learning
● Familiarity with cloud platforms and services (e.g., AWS, Azure, Google Cloud) for scalable AI/ML deployment
● Solid understanding of software engineering principles and best practices for writing maintainable and scalable code
● Excellent analytical and problem-solving skills, with the ability to think critically and propose innovative solutions
● Effective communication skills to collaborate with cross-functional teams and present complex technical concepts to non-technical stakeholders











