AI/ML Trainee | Generative AI

Job Description:

We are looking for a hands-on AI Engineer with experience in Generative AI and Agentic AI to build and deploy production-ready AI solutions.

Key Responsibilities:

• Develop and deploy GenAI and Agentic AI applications.
• Build RAG pipelines, LLM workflows, and AI agents.
• Develop solutions using Python, LangChain, LangGraph, LlamaIndex, or similar frameworks.
• Implement tool calling, context retrieval, and LLM orchestration.
• Integrate AI solutions with APIs and cloud platforms.
• Work with AWS/Azure/GCP, Docker, and CI/CD.

Required Skills:

• Strong Python programming skills.
• 3+ years of GenAI/Agentic AI experience.
• RAG and LLM orchestration.
• LangChain / LangGraph / LlamaIndex / AutoGen / CrewAI / Semantic Kernel.
• MCP and A2A knowledge.
• Cloud, APIs, Docker, and CI/CD experience.

Preferred Experience:

Hands-on experience building and deploying production-ready AI solutions.

Role Overview

We are looking for an experienced AI/ML Engineer with strong expertise in Python, Generative AI, LLMs, LangChain, and LangGraph. The candidate will be responsible for designing and developing AI-powered applications, intelligent agents, and scalable LLM-based solutions.

Key Responsibilities

• Design and develop AI/ML and Generative AI applications using Python and modern LLM technologies.
• Build LLM-based applications and AI agents using LangChain and LangGraph.
• Develop agentic workflows involving tool calling, memory, reasoning, and multi-step orchestration.
• Integrate LLMs such as OpenAI, Azure OpenAI, Anthropic, Gemini, or other foundation models.
• Develop RAG (Retrieval-Augmented Generation) pipelines using vector databases.
• Work with embeddings, prompt engineering, semantic search, and document processing.
• Develop scalable APIs and backend services using Python, FastAPI, or Flask.
• Build and integrate AI solutions with existing applications and enterprise systems.
• Deploy and maintain AI/ML solutions on cloud platforms.
• Collaborate with data scientists, software engineers, and product teams to develop business-focused AI solutions.

Required Skills

• Strong hands-on experience in Python programming.
• Strong experience in AI/ML and Generative AI.
• Hands-on experience with LLMs and LLM-based application development.
• Strong experience with LangChain and/or LangGraph.
• Experience building AI Agents / Agentic AI workflows.
• Strong understanding of RAG, embeddings, vector databases, and prompt engineering.
• Experience with vector databases such as FAISS, Pinecone, Chroma, Weaviate, or Azure AI Search.
• Experience developing REST APIs using FastAPI/Flask.
• Good understanding of Machine Learning, NLP, and deep learning concepts.
• Experience with Azure / AWS / GCP cloud platforms.

Good to Have

• Experience with multi-agent systems and agent orchestration.
• Knowledge of MLOps / LLMOps.
• Experience with Docker, Kubernetes, and CI/CD.
• Knowledge of LLM evaluation, monitoring, observability, and AI governance.
• Experience with Azure OpenAI, Azure AI Foundry, or AWS Bedrock.

Hiring for Junior AI Engineer

Exp : 4 - 6 yrs

Edu : BE/B.Tech/MCA

Work Location : Pune WFO

Skills :

- Min 3 years strong programming experience in Python is a MUST

- Min 2 years hands-on experience in AI with LLMs, RAG pipelines, and AI frameworks

- Experience with cloud platforms (AWS/Azure/GCP)

Hiring for AI Engineer

Exp: 5 - 12 yrs

Edu : BE/B.Tech/MCA

Work Location : Pune / Mumbai

Skill Set:

Total experience ranging from 5–10 years in software engineering/AI roles

Min 5 years strong programming experience in Python or Typescript is a MUST

Min 2.5 years hands-on experience in AI with LLMs, RAG pipelines, and AI frameworks

2+ years shipping LLM systems in production

Experience with cloud platforms (AWS/Azure/GCP)

About the role

We are building AI systems that read, understand and act on real business documents, bank statements, financial reports, policy documents and forms and putting them into production where accuracy and cost both matters.

This is not a research role and it is not a prompt-writing role. You will own features end to end: pick and deploy open-source models, build the pipelines around them, measure whether they actually work on our documents, drive the cost per document down, and keep the whole thing running in production.

You will work closely with the engineering and product teams, and your work will be directly used by business users from day one.

What you will do

Deploy and evaluate open-source models

• Select, deploy and benchmark open-source LLMs and vision-language models for specific, narrow use cases not general chat.
• Build evaluation sets from real documents and define what "good" means numerically (field-level accuracy, extraction recall, hallucination rate) before shipping.
• Run structured comparisons between models and approaches, and write up the trade-offs so the team can make a decision.
• Apply quantization, batching and other optimizations to fit models into a sensible GPU budget.

Build and optimize AI orchestration

• Design multi-step pipelines that combine deterministic code, ML models and LLM calls and know when not to use an LLM.
• Optimize for latency, cost and reliability: caching, batching, request routing, fallback tiers, retries and graceful degradation.
• Instrument pipelines so failures are visible and traceable rather than silent.

Ship to production

• Package models and services with Docker, expose them behind clean APIs, and deploy them to our GPU and CPU infrastructure.
• Handle the unglamorous production concerns: cold starts, timeouts, concurrency limits, versioning, rollback and monitoring.
• Own on-call-style responsibility for the AI features you build, including cost tracking.

Must-have skills

Programming & engineering

• Strong Python: type hints, async/await, dataclasses/Pydantic, clean module design, testing.
• REST API development with FastAPI (or Flask/Django with a willingness to move to FastAPI).
• Git, code review discipline, and the ability to write code someone else can maintain.
• Comfortable in Linux and on the command line.

Machine learning fundamentals

• Working knowledge of PyTorch and the Hugging Face ecosystem (transformers, tokenizers, accelerate).
• Understanding of inference-time concepts: tokenization, context windows, batching, precision (FP16/BF16/INT8), memory footprint.
• Ability to read a model card and a paper well enough to judge whether a model fits a use case.

Document processing

• Hands-on experience with at least two of: pypdfium2, PyMuPDF, pdfplumber, pdfminer.six, Docling, Unstructured, Surya, DocTR, LayoutLM family.
• Practical OCR experience (Tesseract, PaddleOCR, or a cloud OCR) and an understanding of when OCR is the wrong tool.
• Experience extracting tables from PDFs and dealing with merged cells, multi-line rows, and inconsistent column layouts.

Strongly preferred

You will be a much stronger candidate with any of these. We do not expect all of them.

Model serving & optimization

• vLLM, TGI, Ollama, llama.cpp, or Triton Inference Server.
• Quantization formats and tooling: GGUF, AWQ, GPTQ, bitsandbytes, ONNX Runtime, INT8 export.
• Serverless GPU platforms: Modal, RunPod, Replicate, Baseten including cold-start and container-lifecycle management.
• LoRA / QLoRA fine-tuning with PEFT for narrow, task-specific improvements.

Vision-language models

• Practical use of open VLMs: Qwen2.5-VL, InternVL, Granite Vision, Molmo, Phi-Vision, or similar.
• Awareness of where VLMs hallucinate especially on numeric and financial content and patterns for constraining them (using the model for layout only, sourcing values from the text layer, constrained decoding).

Orchestration & pipelines

• Workflow orchestration: Dagster, Airflow, Prefect, or Temporal.
• Async job patterns: Celery, RQ, or platform-native spawn/poll patterns.
• LLM orchestration frameworks (LangGraph, LlamaIndex, Haystack) with the judgement to know when plain Python is a better answer.
• Structured output enforcement: Instructor, Outlines, XGrammar, JSON schema / tool-use modes.

Evaluation & observability

• Building golden datasets and regression suites for extraction tasks.
• Eval tooling: promptfoo, DeepEval, Ragas, or in-house harnesses.
• LLM tracing and monitoring: Langfuse, Arize Phoenix, LangSmith, OpenTelemetry.

Nice extras

• Rule engines and policy evaluation (Open Policy Agent / Rego, Drools, rule-engine).
• Experience in fintech, lending, insurance or accounting documents.
• Handling of PII and data-security practices in document pipelines.
• Contributions to open-source ML or document-processing projects.

Why join us

• Real production ownership from month one your work goes to actual users, not a demo.
• Genuinely hard technical problems in document AI, not wrappers over an API.
• Small team, short decision cycles, direct access to leadership.
• Budget and freedom to evaluate and adopt new open-source models as they land.

To apply: send your CV along with a short note on one AI system you have taken to production what it did, what the accuracy was, and what broke.

Support with design and build to prove out agentic AI solution flow by working with other data 

scientists and engineers to build, train Large Language Model (LLM) architectures, RAG 

systems, and autonomous agentic workflows 

Key qualifications: 

>> AI solution design & Development: Design Agentic AI solutions using RAG (Retrieval-

Augmented Generation) and orchestration frameworks like LangGraph or LangChain. 

>> Model Fine-Tuning: Solid understanding and experience with Pre-train, fine-tune, and 

optimize open-source like BERT, LLama, and other proprietary foundation models for domain-

specific tasks 

>> Solid Stats and ML foundations and (vibe) coding skills with Python, PySpark 

>>  Implement validation frameworks and tracing practices (using tools like Arize) to monitor 

agent behavior, guard against model drift, and ensure compliance 

>> Collaborate with Engineering to deploy models securely on cloud and on-prem ecosystems 

About the Role

We are looking for enthusiastic LLM Interns to join our team remotely for a 3-month internship. This role is ideal for students or graduates interested in AI, Natural Language Processing (NLP), and Large Language Models (LLMs). You will gain hands-on experience working with cutting-edge AI tools, prompt engineering, and model fine-tuning. While this is an unpaid internship, interns who successfully complete the program will receive a Completion Certificate and a Letter of Recommendation.

Responsibilities

• Research and experiment with LLMs, NLP techniques, and AI frameworks.
• Design, test, and optimize prompts and workflows for different use cases.
• Assist in fine-tuning or integrating LLMs for internal projects.
• Evaluate model outputs and improve accuracy, efficiency, and reliability.
• Collaborate with developers, data scientists, and product managers to implement AI-driven features.
• Document experiments, results, and best practices.

Requirements

• Strong interest in Artificial Intelligence, NLP, and Machine Learning.
• Familiarity with Python and ML libraries (e.g., TensorFlow, PyTorch, Hugging Face Transformers).
• Basic understanding of LLM concepts such as embeddings, fine-tuning, and inference.
• Knowledge of APIs (OpenAI, Anthropic, Hugging Face, etc.) is a plus.
• Good analytical and problem-solving skills.
• Ability to work independently in a remote environment.

What You’ll Gain

• Practical exposure to state-of-the-art AI tools and LLMs.
• Mentorship from AI and software professionals.
• Completion Certificate upon successful completion.
• Letter of Recommendation based on performance.
• Experience to showcase in research projects, academic work, or future AI roles.

Internship Details

• Duration: 3 months
• Location: Remote (Work from Home)
• Stipend: Unpaid
• Perks: Completion Certificate + Letter of Recommendation

Job Title: Python + GenAI Developer

Experience: 4–6 Years

Location: Bangalore / Hyderabad (as applicable)

Job Description:

We are looking for a skilled Python + Generative AI Developer with hands-on experience in Python development, LLM integration, and building AI-powered applications.

Key Responsibilities:

• Develop backend applications and REST APIs using Python.
• Build and integrate Generative AI solutions using LLMs.
• Work with frameworks such as LangChain or LangGraph.
• Implement Retrieval-Augmented Generation (RAG) pipelines and vector databases.
• Integrate OpenAI or other LLM APIs into applications.
• Collaborate with teams to develop, test, and deploy AI-powered solutions.
• Work with cloud platforms such as AWS, Azure, or GCP.

Required Skills:

• 4–6 years of Python development experience.
• Hands-on experience with Generative AI and LLMs.
• Experience with LangChain or LangGraph.
• Knowledge of RAG and vector databases.
• Experience developing REST APIs using FastAPI or Flask.
• Understanding of prompt engineering and LLM integration.
• Familiarity with Git and cloud platforms.

Preferred Skills:

• Experience with Agentic AI and AI agents.
• Knowledge of embeddings and semantic search.
• Experience with Docker and CI/CD.

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.

Role Overview

We are looking for a Python Developer with strong experience in Generative AI and LLM-based applications. The candidate should have hands-on experience building AI solutions using Python, RAG, LangChain/LangGraph, and related GenAI technologies.

Mandatory Skills

Python, GenAI/LLM, RAG, LangChain/LangGraph, Agentic AI, FastAPI, REST API, Vector Database, Prompt Engineering, Microservices

Key Responsibilities

• Develop and maintain applications using Python and modern frameworks.
• Build GenAI/LLM-based applications and solutions.
• Develop RAG pipelines using vector databases.
• Work with LangChain/LangGraph for LLM and agent-based applications.
• Develop and integrate REST APIs using FastAPI.
• Implement Agentic AI workflows and AI-powered features.
• Integrate LLMs with existing applications and microservices.
• Apply prompt engineering techniques to improve AI application performance.