Talent.com
AI Engineer / Senior AI Engineer – Document Intelligence
AI Engineer / Senior AI Engineer – Document IntelligencePiazza Consulting Group • India
AI Engineer / Senior AI Engineer – Document Intelligence

AI Engineer / Senior AI Engineer – Document Intelligence

Piazza Consulting Group • India
10 hours ago
Job description

We are looking for someone obsessed with turning messy real-world documents into perfectly structured, actionable data.

If you live and breathe document layout analysis, OCR post-processing, and visual + language models, this role is custom-built for you.

Core Responsibilities

  • Design and build production-grade Document Intelligence pipelines (invoices, contracts, forms, reports, handwritten, tables, multi-language, etc.)
  • Train / fine-tune and deploy Layout-aware models (LayoutLMv3, Donut, LLaMA-Adapter, Nougat, etc.)
  • Build and optimize Vision-Language models (VLLMs) on custom enterprise document datasets
  • Improve OCR accuracy using layout context, post-correction with LLMs, and geometric reasoning
  • Own the full stack : data preparation → model training (PyTorch) → evaluation → ONNX / TensorRT optimization → FastAPI deployment
  • Push boundaries on table extraction, key-value pairing, nested hierarchies, and multi-page document understanding

Must-Have Skills & Experience

  • Very strong understanding of document layout analysis (bounding boxes, reading order, logical blocks, nested tables, headers / footers, multi-column detection)
  • Hands-on experience with modern Document AI architectures :
  • LayoutLMv1 / v2 / v3, DocFormer, LayoutXLM, Donut, Pix2Struct, Nougat, UDOP, etc.
  • Vision-Language models (LLaVA, Qwen-VL, InternVL, PaliGemma, etc.)
  • Deep experience fine-tuning and serving LLMs & VLLMs (Llama-3, Mistral, Phi-3-vision, Qwen, etc.) using PEFT (LoRA / QLoRA), vLLM, TGI, or Ollama
  • Strong PyTorch proficiency (custom trainers, distributed training with DDP / FSDP, TorchCompile, mixed precision)
  • Solid grasp of OCR ecosystems and post-processing (Tesseract, EasyOCR, PaddleOCR, AWS Textract / Google Document AI limitations and how to beat them)
  • Experience building datasets from real enterprise documents (Labelling tools : UBIAI, Label Studio, Doccano, custom UI)
  • Good applied math : Transformers, attention mechanisms, positional encodings (especially 2D layouts), RoPE, ALiBi
  • Nice-to-Have (Big Bonus)

  • Published research or open-source contributions in Document AI / VLLM space
  • Experience with multimodal RAG over documents
  • ONNX / TensorRT / DeepSpeed optimization for low-latency inference
  • Kubernetes + GPU scheduling (we run our own bare-metal cluster)
  • Who thrives here?

    You get excited when you see a 50-page scanned purchase order with overlapping stamps and handwritten notes — because you already know exactly how you’re going to destroy it.

    Perks

  • Work directly on enterprise deals worth crores — your model = real revenue impact
  • Unlimited GPU access (A100s & H100s in-house)
  • Create a job alert for this search

    Senior Ai Engineer • India

    Related jobs
    Senior Applied AI Engineer

    Senior Applied AI Engineer

    SmarTek21 • Republic Of India, IN
    We are looking for a highly skilled AI Engineer to build and scale enterprise-grade AI solutions using advanced LLMs, agent frameworks, and automation platforms. In this hands-on role, you will desi...Show more
    Last updated: 22 days ago • Promoted
    Senior Applied AI Engineer

    Senior Applied AI Engineer

    Luxoft • Republic Of India, IN
    Our client, one of the leading Agriculture Companies, is modernising their landscape and adopting AI and innovations in their process. We are seeking a highly skilled and innovative Python / AI Engine...Show more
    Last updated: 19 days ago • Promoted
    Senior AI Engineer

    Senior AI Engineer

    IBS Software • India
    Trivandrum / Kochi / Chennai / Bangalore.We are seeking a talented AI Engineer to join our Data & AI Center of Excellence. You will join to build AI-first capabilities across IBS products and customer sol...Show more
    Last updated: 1 day ago • Promoted
    Senior AI / ML Cloud Engineer – Graph Analytics Platform

    Senior AI / ML Cloud Engineer – Graph Analytics Platform

    ContexQ • India, India
    Senior AI / ML Cloud Engineer – Graph Analytics Platform.ContexQ, a Singapore-based B2B SaaS AI startup, is dedicated to transforming financial crime, fraud, and risk management through a groundbre...Show more
    Last updated: 5 days ago • Promoted
    Senior AI / ML Engineer

    Senior AI / ML Engineer

    Luxoft • India, India
    Our client, one of the leading Agriculture Companies, is modernising their landscape and adopting AI and innovations in their process. We are seeking a highly skilled and innovative Python / AI Engine...Show more
    Last updated: 19 days ago • Promoted
    Senior AI Solutions Engineer

    Senior AI Solutions Engineer

    Litmus7 • Cochin, Republic Of India, IN
    As part of this initiative, resource should research and experiment with the latest AI and cloud innovations (such as AWS Agents, Databricks AI, and other Model Context Protocol (MCP integrations),...Show more
    Last updated: 22 days ago • Promoted
    Senior AI Platform Engineer

    Senior AI Platform Engineer

    Cargill • India
    Platform Engineer for AI Assistants & Agents in AI & Data Science.Cargills portfolio of GenAI Assistants and Agents - including ChatGPT-powered copilots and bespoke LLM services.You will translate ...Show more
    Last updated: 2 days ago • Promoted
    AI Innovation and Integration Engineer

    AI Innovation and Integration Engineer

    Litmus7 • Cochin, Republic Of India, IN
    As part of this initiative, resource should research and experiment with the latest AI and cloud innovations (such as AWS Agents, Databricks AI, and other Model Context Protocol (MCP integrations),...Show more
    Last updated: 22 days ago • Promoted
    Senior AI Engineer

    Senior AI Engineer

    Wellness Coach • India, India
    Our mission is to help millions of people live healthier, more mindful, and more balanced lives — powered by technology, compassion, and science. We’re building next-generation.You’ll research, arch...Show more
    Last updated: 30+ days ago • Promoted
    Senior AI / ML Engineer

    Senior AI / ML Engineer

    RingCentral • India
    We are seeking an experienced AI Engineer with a strong background in Natural Language Understanding (NLU) who is passionate about pushing the boundaries of Conversational AI.In this role, you will...Show more
    Last updated: 3 days ago • Promoted
    Applied AI Engineer

    Applied AI Engineer

    Strategic Talent Partner • India, India
    Design and deploy advanced multi-agent pipelines for credit analysis.Optimize inference and prompt chains using frameworks like DSPy, GEPA, and LangChain. Implement reasoning techniques (CoT, ToT, G...Show more
    Last updated: 23 days ago • Promoted
    Senior AI Engineer

    Senior AI Engineer

    NextDimension AI • India
    INR 12-30 LPA Base + Bonus + Equity.US-based technology startup building AI Agents in Healthcare, established by a team of distinguished AI / ML Scientists and Engineers from.We're empowering Enterpr...Show more
    Last updated: 3 days ago • Promoted
    Senior Ai Engineer

    Senior Ai Engineer

    SmarTek21 • Republic Of India, IN
    We are looking for a highly skilled AI Engineer to build and scale enterprise-grade AI solutions using advanced LLMs, agent frameworks, and automation platforms. In this hands-on role, you will desi...Show more
    Last updated: 22 days ago • Promoted
    Senior Software Engineer - AI

    Senior Software Engineer - AI

    Deep Cognition • India
    Large Language Model (LLM) APIs.OpenAI, Anthropic, or similar frameworks.The ideal candidate will have a strong background in . API integration, and workflow automation, with a passion for building ...Show more
    Last updated: 3 days ago • Promoted
    Senior AI Platform Engineer

    Senior AI Platform Engineer

    Ocrolus • Republic Of India, IN
    Come build at the intersection of AI and fintech.At Ocrolus, we’re on a mission to help lenders automate workflows with confidence—streamlining how financial institutions evaluate borrowers and ena...Show more
    Last updated: 22 days ago • Promoted
    Senior Applied AI Engineer

    Senior Applied AI Engineer

    Litmus7 • Cochin, Republic Of India, IN
    As part of this initiative, resource should research and experiment with the latest AI and cloud innovations (such as AWS Agents, Databricks AI, and other Model Context Protocol (MCP integrations),...Show more
    Last updated: 22 days ago • Promoted
    Senior AI Engineer

    Senior AI Engineer

    Kanerika Inc • India
    We are seeking an experienced Senior AI Engineer with 3+years of experience to join our dynamic team, driving innovation in AI-driven solutions with a global impact. This role is specifically tailor...Show more
    Last updated: 3 days ago • Promoted
    Senior AI / ML Engineer

    Senior AI / ML Engineer

    NextDimension AI • India
    INR 12-30 LPA Base + Bonus + Equity.NextDimension is a US-based technology startup building AI Agents in Healthcare, established by a team of distinguished AI / ML Scientists and Engineers from Googl...Show more
    Last updated: 3 days ago • Promoted