Senior Machine Learning Engineer

Roles & Responsibilities

• Design and develop intelligent AI-based applications using advanced NLP and LLM techniques to solve real-world business challenges in financial services.
• Build and optimize Retrieval-Augmented Generation (RAG) pipelines leveraging structured and unstructured financial data.
• Integrate and orchestrate LLMs/SLMs for question-answering, summarization, semantic search, and document understanding.
• Develop and maintain RESTful APIs (sync and async) to serve NLP models and chatbot interfaces using frameworks like FastAPI, Flask, etc.
• Should have knowledge of advanced prompting techniques.
• Implement semantic search, hybrid search, and text retrieval systems using Elasticsearch and vector databases (e.g., FAISS, Pinecone, Weaviate).
• Perform NLP tasks such as entity recognition, text classification, intent detection, embedding generation, and sentiment analysis where required.
• Monitor and fine-tune LLM/SLM performance with real-world user data to improve relevance, latency, and accuracy.
• Exposure to LLMOps tools for monitoring, evaluation, and versioning of AI models in production.
• Build, train, and evaluate deep learning models for NLP tasks including classification, NER, summarization, and embedding generation.
• Develop traditional machine learning models (e.g., regression, decision trees, clustering) for structured data analysis and prediction tasks.
• Interact with cross-functional teams to understand system issues and follow up with respective teams to get them fixed.
• Understand and identify areas of improvement across businesses and participate in solution identification and implementation.
• Should be able to work as an Individual Contributor on new and existing projects.
• Positive and problem-solving attitude, must work as an independent contributor.

Ideal Candidate

1.Strong Data Scientist / AI Engineer / Generative AI Engineer profile.

2.Mandatory (Experience 1) - Must have 3+ years of hands-on experience in Data Science, Artificial Intelligence, Machine Learning, Deep Learning, NLP, or Generative AI application development.

3.Mandatory (Experience 2) - Must have strong hands-on experience in Python programming, backend development, API development, and production-grade application support

4.Mandatory (Experience 3) - Must have experience working with Machine Learning and Deep Learning frameworks such as PyTorch, TensorFlow, Keras, or Scikit-learn.

5.Mandatory (Experience 4) - Must have hands-on experience in NLP use cases such as text classification, sentiment analysis, entity recognition (NER), semantic search, embeddings, or document understanding.

6.Mandatory (Experience 5) - Must have experience working with Large Language Models (LLMs) such as GPT, LLaMA, Mistral, Phi, Claude, Gemini, or similar models

.7.Mandatory (Experience 6) - Must have hands-on experience building or implementing Retrieval Augmented Generation (RAG) solutions, vector search, semantic search, or knowledge-based AI applications.

8.Mandatory (Experience 7) - Must have experience with Prompt Engineering and Generative AI frameworks such as LangChain, LangGraph, AI Agents, Azure OpenAI, or similar technologies.

9.Mandatory (Experience 8) - Must have experience developing, consuming, or integrating APIs using Python frameworks such as FastAPI, Flask, or similar technologies.

10.Mandatory (CTC) - The CTC breakup offered will be 75% fixed + 25% variable, as per company policy.

11.Preferred (Experience 1) - Experience with LLMOps/MLOps tools for monitoring, evaluation, experimentation, and versioning of AI models.

12.Preferred (Experience 2) - Exposure to Azure OpenAI, Azure Kubernetes Service (AKS), Kubernetes, cloud-native AI deployments, or distributed systems.

13.Preferred (Experience 3) - Experience working with PostgreSQL, MongoDB, Redis, Kafka, or large-scale data platforms.

14.Preferred (Experience 4) - Familiarity with Docker, Kubernetes, cloud platforms, and scalable deployment architecture.

15.Preferred (Company) - Candidates from AI-first startups, product companies, SaaS organizations, fintech, or data-driven technology companies.

16.Mandatory ( Age ) - Candidate Should be Below 28 Years.

17.Mandatory ( Pedigree) - B.TECH / M.TECH from Tier 1 Colleges (IIT's, NIT's, BITS) are Considered.

Strong Data Scientist / AI Engineer / Generative AI Engineer profile.

2

Mandatory (Experience 1) - Must have 3+ years of hands-on experience in Data Science, Artificial Intelligence, Machine Learning, Deep Learning, NLP, or Generative AI application development.

3

Mandatory (Experience 2) - Must have strong hands-on experience in Python programming, backend development, API development, and production-grade application support.

4

Mandatory (Experience 3) - Must have experience working with Machine Learning and Deep Learning frameworks such as PyTorch, TensorFlow, Keras, or Scikit-learn.

5

Mandatory (Experience 4) - Must have hands-on experience in NLP use cases such as text classification, sentiment analysis, entity recognition (NER), semantic search, embeddings, or document understanding.

6

Mandatory (Experience 5) - Must have experience working with Large Language Models (LLMs) such as GPT, LLaMA, Mistral, Phi, Claude, Gemini, or similar models.

7

Mandatory (Experience 6) - Must have hands-on experience building or implementing Retrieval Augmented Generation (RAG) solutions, vector search, semantic search, or knowledge-based AI applications.

8

Mandatory (Experience 7) - Must have experience with Prompt Engineering and Generative AI frameworks such as LangChain, LangGraph, AI Agents, Azure OpenAI, or similar technologies.

9

Mandatory (Experience 8) - Must have experience developing, consuming, or integrating APIs using Python frameworks such as FastAPI, Flask, or similar technologies.

10

Mandatory (CTC) - The CTC breakup offered will be 75% fixed + 25% variable, as per company policy.

11

Preferred (Experience 1) - Experience with LLMOps/MLOps tools for monitoring, evaluation, experimentation, and versioning of AI models.

12

Preferred (Experience 2) - Exposure to Azure OpenAI, Azure Kubernetes Service (AKS), Kubernetes, cloud-native AI deployments, or distributed systems.

Strong AI Engineer / Machine Learning Engineer profiles.

2

Mandatory (Experience 1) – Must have minimum 3+ years of hands-on experience in Data Science, Machine Learning, Applied AI, NLP, Deep Learning, or Generative AI solutions.

3

Mandatory (Experience 2) – Must have strong hands-on experience in Python programming, SQL, data analysis, feature engineering, model development, and production-grade ML applications.

4

Mandatory (Experience 3) – Must have experience working with Machine Learning and Deep Learning frameworks such as PyTorch, TensorFlow, Keras, Scikit-learn, or equivalent.

5

Mandatory (Experience 4) – Must have hands-on experience working on NLP, embeddings, semantic search, text classification, document understanding, recommendation systems, or similar AI/ML use cases.

6

Mandatory (Experience 5) – Must have experience working with Large Language Models (LLMs) such as GPT, Llama, Mistral, Claude, Gemini, Phi, or similar foundation models.

7

Mandatory (Experience 6) – Must have hands-on experience building or implementing RAG (Retrieval Augmented Generation) systems, vector search, knowledge retrieval, embeddings, chunking, indexing, or semantic retrieval solutions.

8

Mandatory (Experience 7) – Must have experience working with Git, CI/CD practices, production environments, and scalable AI/ML systems.

9

Mandatory (CTC) – The CTC breakup offered will be 75% fixed + 25% variable, as per company policy.

10

Mandatory (Age) - Candidate's Age should be below 28 Years

This is a remote position.

About Leegality:

Leegality works with large Indian businesses to digitally transform critical compliance processes in a fast, easy and secure way.

We have multiple products across 2 categories:

Document Infrastructure:

Products that help businesses build paperless processes at scale:

1. Document Execution Workflow: A unified platform for businesses to digitally execute (eSign, eStamp, Template Pre-fill, Document Fraud Prevention etc.) agreements, forms and other documents in a compliant way. Currently in use by 2000+ Indian businesses from giants like HDFC and SBI Cards to high-growth disruptors like goDigit and Cars24.
2. Contract Management: An AI-powered platform for businesses to quickly review, negotiate and take action on contract
3. Signstation: A simple platform for businesses to digitally sign simple documents like invoices, policies and letters in a cost effective manner

Consent Infrastructure:

1. Consentin: An end-to-end DPDP and Privacy compliance platform for Indian businesses
2. Consentin Lens: A data discovery platform for businesses to identify the personal data they collect and store.

If you’re interested in building mission critical software that operates at population scale (75 million + Indians have signed at least one document through Leegality) then join Leegality.

Curious about our impact? Explore our customer success stories: leegality.com/case-studies

Our Culture

At Leegality, trust, ownership, transparency, and having fun while doing meaningful work are core to how we operate — not just values on paper. Our team rated us an incredible 97 eNPS for FY 2023–24 — the highest among 175+ startups surveyed.

We focus deeply on helping our people grow and stay motivated. Some of the perks you’ll enjoy:

• Flexible working hours
• Hybrid work setup
• Bi-annual performance appraisals
• A culture that rewards initiative, curiosity, and impact

If you're looking for a place where you can make a real difference while working with smart, driven, and genuinely nice people, welcome to Leegality.

Location: Hybrid

Job Brief:

• As a Machine Learning Engineer specializing in Computer Vision (CV) and Natural Language Processing (NLP), you will develop solutions to interesting technical problems, exploring exciting growth opportunities and having a real impact on our product, particularly focusing on document and content intelligence.
• To ensure success, you should demonstrate solid data science knowledge and experience in a related ML, CV, or NLP role. A first-class engineer will be someone whose expertise enhances our systems for document intelligence and content processing

Responsibilities:

• Designing machine learning systems, self-running artificial intelligence (AI) software, and specialized models for Computer Vision and Natural Language Processing applications.
• Transforming data science prototypes and applying appropriate deep learning algorithms and tools to text and image/document data.
• Solving complex CV and NLP problems with multi-layered data types, such as image/document classification, information extraction, semantic search, and object detection.
• Optimizing existing machine learning models, with a focus on high-performance model deployment for CV and NLP tasks.
• Developing ML algorithms (including large language models/LLMs and computer vision models) to analyze huge volumes of historical text, image, and document data to make predictions and automate workflows.
• Running tests, performing statistical analysis, and interpreting test results for CV/NLP model performance.
• Documenting machine learning processes, model architectures, and data pipelines.
• Keeping abreast of developments in machine learning, Computer Vision, and Natural Language Processing.

Requirements:

• 3+ years of relevant experience in Machine Learning Engineering, with a strong focus on Computer Vision and/or Natural Language Processing.
• Advanced proficiency with Python.
• Extensive knowledge of ML frameworks, libraries (e.g., PyTorch, Transformers), data structures, data modeling, and software architecture.
• Experience with building and maintaining scalable RESTful APIs (e.g., FastAPI).
• In-depth knowledge of mathematics, statistics, deep learning (CNNs, RNNs, Transformers), and algorithms.
• Superb analytical and problem-solving abilities, especially for unstructured data challenges.
• Great communication and collaboration skills.
• Excellent time management and organizational abilities.
• Experience with cloud platforms (e.g., AWS) for model deployment and MLOps.

Recruitment Process:

• Our hiring process combines AI-powered evaluations with structured interviews to ensure a fair and seamless experience.
• You will be contacted via email with the next steps upon being shortlisted.
• The process may include Assessments, AI-enabled interviews, and In-Person Interviews with our team.
• Final selection and CTC will be based on your overall performance and experience.

Apply directly through our career page: https://careers.leegality.com/jobs/Careers

For more information about us please visit our:

Our Company and Culture: https://bit.ly/3Iqm5SB

Our Website: www.leegality.com/

Our LinkedIn Page: www.linkedin.com/company/leegality/

Leegality's Privacy Notice: https://www.leegality.com/employee-privacy-notice

Job Summary

We are looking for an experienced AI/ML Engineer to design, develop, deploy, and maintain machine learning and AI solutions that address complex business problems. The ideal candidate should have strong hands-on experience in Python, Machine Learning, Generative AI, LLMs, and AI/ML deployment, with the ability to work across the complete AI/ML lifecycle.

Key Responsibilities

• Design, develop, and deploy scalable Machine Learning and AI models for real-world business use cases.
• Build and optimize ML pipelines covering data preparation, feature engineering, model development, evaluation, and deployment.
• Develop solutions using Generative AI, Large Language Models (LLMs), NLP, and deep learning.
• Work with LLMs, prompt engineering, embeddings, vector databases, and Retrieval-Augmented Generation (RAG) architectures.
• Integrate AI/ML models with enterprise applications and APIs.
• Fine-tune and evaluate ML/LLM models based on business requirements.
• Implement MLOps practices for model versioning, deployment, monitoring, and continuous improvement.
• Collaborate with Data Scientists, Software Engineers, Architects, Product Managers, and business stakeholders.
• Conduct model performance evaluation, optimization, and troubleshooting.
• Ensure AI solutions meet requirements around security, scalability, reliability, responsible AI, and data privacy.
• Stay current with emerging AI/ML technologies, frameworks, and industry best practices.

Required Skills

• Strong programming experience in Python.
• Strong understanding of Machine Learning algorithms, statistics, and data structures.
• Hands-on experience with ML frameworks such as TensorFlow, PyTorch, Scikit-learn, or equivalent.
• Experience with Generative AI and LLMs such as OpenAI, Azure OpenAI, Claude, Gemini, or open-source models.
• Strong knowledge of Prompt Engineering, RAG, embeddings, vector databases, and AI agents.
• Experience with NLP, deep learning, or computer vision is an advantage.
• Experience developing and consuming REST APIs and microservices.
• Working knowledge of SQL and NoSQL databases.
• Experience with cloud platforms such as Azure, AWS, or GCP.
• Understanding of Docker, Kubernetes, CI/CD, and MLOps.
• Familiarity with Git and modern software development practices.

Preferred Qualifications

• Bachelor's or Master's degree in Computer Science, Artificial Intelligence, Data Science, Engineering, or a related field.
• 4 years of relevant experience in AI/ML engineering or a related field.
• Experience building and deploying production-grade AI/ML solutions.
• Enterprise application development experience.
• Experience with Azure OpenAI, AWS Bedrock, Vertex AI, or similar managed AI platforms.
• Experience with LangChain, LlamaIndex, Semantic Kernel, or comparable AI frameworks is a plus.
• Experience with AI/ML model monitoring, evaluation, and optimization.

What You Bring

• Strong problem-solving and analytical skills.
• Ability to translate business requirements into practical AI/ML solutions.
• Strong software engineering and debugging capabilities.
• Ability to work independently as well as collaboratively in a cross-functional environment.
• Good communication skills with the ability to explain complex AI concepts to technical and non-technical stakeholders.

Keywords

AI Engineer | ML Engineer | Machine Learning | Generative AI | LLM | Python | NLP | Deep Learning | RAG | Prompt Engineering | AI Agents | Azure OpenAI | AWS Bedrock | MLOps | TensorFlow | PyTorch | Scikit-learn | Vector Database | Cloud AI

Location: Gurugram

Work mode: Hybrid

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

Job Title : Data Scientist – NLP & LLM

Experience Required : 4 to 5+ Years

Location : Gurugram

Notice Period : Immediate Joiner Preferred

Employment Type : Full-Time

Job Summary :

We are seeking a highly skilled Data Scientist with strong expertise in Natural Language Processing (NLP) and modern deep learning techniques. The ideal candidate will have hands-on experience working with NER, RNN, LSTM, Transformers, GPT models, and Large Language Models (LLMs), including frameworks such as spaCy.

Mandatory Skills : NLP, spaCy, NER (Named-entity Recognition), RNN (Recurrent Neural Network), LSTM (Long Short-Term Memory), Transformers, GPT, LLMs, Python

Key Responsibilities :

• Design and implement NLP models for tasks like text classification, entity recognition, summarization, and question answering.
• Develop and optimize models using deep learning architectures such as RNN, LSTM, and Transformer-based models.
• Fine-tune or build models using pre-trained LLMs such as GPT, BERT, etc.
• Work with tools and libraries including spaCy, Hugging Face Transformers, and other relevant frameworks.
• Perform data preprocessing, feature extraction, and training pipeline development.
• Evaluate model performance and iterate with scalable solutions.
• Collaborate with engineering and product teams to integrate models into production.

Required Skills :

• 4 to 5+ years of hands-on experience in Data Science and NLP.
• Strong understanding of NER, RNN, LSTM, Transformers, GPT, and LLM architectures.
• Experience with spaCy, TensorFlow, PyTorch, and Hugging Face Transformers.
• Proficient in Python and its ML ecosystem (NumPy, pandas, scikit-learn, etc.).
• Familiarity with prompt engineering and fine-tuning LLMs is a plus.
• Excellent problem-solving and communication skills.

Role Overview 

As a Data Scientist, you will work with business stakeholders, AI engineers, and domain experts to transform data into actionable insights and intelligent solutions. You will develop machine learning models, perform statistical analysis, and contribute to AI-driven products that create measurable business impact. 

Key Responsibilities 

Data Science & Machine Learning 

• Analyze structured and unstructured data to identify patterns, trends, and business opportunities.  
• Perform exploratory data analysis (EDA), feature engineering, and data preparation.  
• Develop, evaluate, and optimize machine learning models for prediction, classification, clustering, and forecasting.  
• Apply statistical techniques to solve business problems and validate model performance.  
• Design and execute experiments to improve model accuracy and business outcomes.  

 AI Solution Development 

• Collaborate with AI Engineers, Data Engineers, and domain experts to build AI-powered solutions.  
• Translate business requirements into scalable data science approaches.  
• Contribute to Generative AI and advanced analytics initiatives where applicable.  
• Document methodologies, model performance, and key findings.  

 Required Technical Skills 

• Strong programming skills in Python and SQL for data analysis, feature engineering, and machine learning.  
• Strong understanding of Statistics, Probability, Linear Algebra, and Calculus as applied to machine learning and data science.  
• Experience with Exploratory Data Analysis (EDA), data preprocessing, feature engineering, feature selection, and handling missing or imbalanced data.  
• Good understanding of Supervised, Unsupervised, and Ensemble Machine Learning algorithms, including their assumptions, strengths, limitations, and appropriate use cases.  
• Strong knowledge of Regression, Classification, Clustering, Time Series Forecasting, Dimensionality Reduction, Recommendation Systems, and Anomaly Detection techniques.  
• Experience with Model Evaluation, Cross-Validation, Hyperparameter Optimization, Bias-Variance Trade-off, Feature Importance, Explainable AI (XAI), and Performance Metrics.  
• Understanding of Statistical Inference, Hypothesis Testing, Probability Distributions, Sampling Techniques, Confidence Intervals, and A/B Testing.  
• Experience translating business problems into analytical approaches and developing scalable, data-driven solutions.  
• Working knowledge of Generative AI, Large Language Models (LLMs), Prompt Engineering, and Retrieval-Augmented Generation (RAG) is preferred. 

Preferred Qualifications 

• Bachelor's or master's degree in computer science, Artificial Intelligence, Data Science, Statistics, Mathematics, Engineering, or a related field.  
• 2–4 years of experience developing machine learning or data science solutions.  
• Experience working on end-to-end data science projects in a business environment.  

 Nice to Have 

• Exposure to Generative AI, LLMs, RAG, or Agentic AI.  
• Experience with Computer Vision or Natural Language Processing (NLP).  
• Familiarity with cloud-based AI platforms.  
• Knowledge of construction, engineering, manufacturing, or industrial domains.  
• Participation in hackathons, research, Kaggle competitions, or open-source projects.  

 Soft Skills 

Strong analytical and problem-solving skills, effective communication and collaboration, ownership mindset, adaptability, continuous learning, and a passion for innovation.

Job Description – Python & Generative AI Engineer

Python & Generative AI Engineer

Location: Bangalore, India

Experience: 4+ Years

Employment Type: Full-time

Job Summary

We are looking for an experienced Python & Generative AI Engineer with 4+ years of software development experience and strong hands-on expertise in Python, LLMs, Generative AI, and AI application development.

The ideal candidate should be comfortable building production-grade AI solutions, integrating LLMs with enterprise applications, and developing scalable APIs and services using Python.

Key Responsibilities

• Design, develop, and deploy Generative AI applications using Python and modern AI/ML frameworks.
• Work with Large Language Models (LLMs) such as GPT, Claude, Gemini, Llama, or similar models.
• Develop RAG (Retrieval-Augmented Generation) pipelines using vector databases and embedding models.
• Build AI-powered applications using frameworks such as LangChain, LangGraph, LlamaIndex, or equivalent.
• Develop scalable REST APIs and backend services using FastAPI, Flask, or Django.
• Integrate LLM APIs, prompt engineering, function/tool calling, and structured outputs into enterprise applications.
• Work with vector databases such as FAISS, Pinecone, Weaviate, Milvus, or Azure AI Search.
• Implement document processing, chunking, embeddings, semantic search, and knowledge retrieval solutions.
• Evaluate LLM responses for accuracy, relevance, hallucination, latency, and cost.
• Build and maintain production-ready AI pipelines with appropriate monitoring, logging, security, and error handling.
• Collaborate with data scientists, ML engineers, software engineers, and business stakeholders to deliver AI solutions.
• Write clean, reusable, testable, and well-documented Python code.
• Participate in architecture discussions, code reviews, testing, deployment, and production support.

Required Skills

• 4+ years of experience in software/Python development.
• Strong proficiency in Python and object-oriented programming.
• Hands-on experience with Generative AI and LLM-based applications.
• Experience with OpenAI/Azure OpenAI, Anthropic, Google Gemini, or open-source LLMs.
• Strong understanding of Prompt Engineering and LLM application patterns.
• Experience implementing RAG pipelines.
• Knowledge of embeddings, vector databases, semantic search, and document retrieval.
• Experience with LangChain, LangGraph, LlamaIndex, or similar frameworks.
• Strong experience developing REST APIs using FastAPI/Flask/Django.
• Familiarity with Git, CI/CD, Docker, and cloud platforms such as AWS, Azure, or GCP.
• Good understanding of SQL and databases.
• Strong problem-solving and communication skills.

Good to Have

• Experience with AI agents / Agentic AI and tool/function calling.
• Experience with multi-agent frameworks.
• Knowledge of MLOps/LLMOps and model evaluation.
• Experience with Kubernetes and containerized deployments.
• Knowledge of NLP, machine learning, or deep learning.
• Experience with Azure AI, AWS Bedrock, Amazon SageMaker, or Google Vertex AI.
• Understanding of responsible AI, data privacy, and LLM security.

Education

Bachelor's or Master's degree in Computer Science, Information Technology, Engineering, or a related field.

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.