Talent.com
AI Engineer / Senior AI Engineer – Document Intelligence
AI Engineer / Senior AI Engineer – Document IntelligencePiazza Consulting Group • India
No longer accepting applications
AI Engineer / Senior AI Engineer – Document Intelligence

AI Engineer / Senior AI Engineer – Document Intelligence

Piazza Consulting Group • India
16 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: 23 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: 20 days ago • Promoted
    Conversational AI Engineer

    Conversational AI Engineer

    Aonxi.Tech • Republic Of India, IN
    Aonxi is building the world’s first conversion-backed, credit-embedded revenue operating system - where every call, meeting, and conversation gets transcribed, scored, and turned into sales perform...Show more
    Last updated: 2 days 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 Cloud Engineer – Graph Analytics Platform

    Senior Ai / Ml Cloud Engineer – Graph Analytics Platform

    ContexQ • Republic Of India, IN
    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: 6 days ago • Promoted
    AI Engineer / Senior AI Engineer – Document Intelligence

    AI Engineer / Senior AI Engineer – Document Intelligence

    Piazza Consulting Group • India
    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 vis...Show more
    Last updated: 14 hours ago • Promoted • New!
    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
    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 / ML Engineer

    Senior AI / ML Engineer

    Digital Green • Republic Of India, IN
    Job Title : Lead, ML / AI Engineer.Digital Green is a pioneer global not for profit organization, utilizing digital platforms and community-driven approaches to amplify the voices of smallholder farm...Show more
    Last updated: 16 hours ago • Promoted • New!
    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
    Ai Research Senior Engineer

    Ai Research Senior Engineer

    VOICERAcx • Areā, Republic Of India, IN
    Pune, India (Hybrid : 4 days onsite + 1 day remote).We appreciate diverse experiences, but this role is not suitable for candidates whose primary experience is limited to orchestration or prompt eng...Show more
    Last updated: 5 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

    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: 23 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