AI Engineer

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

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.

About the role

We are seeking an AI Engineer to build and implement AI systems for content production at scale. You'll work at the intersection of engineering and content designing prompt pipelines, integrating generative models, and building the tooling that turns source material into finished creative output. The ideal candidate is technically strong but also has taste: someone who understands story and craft, and can tell the difference between output that's technically correct and output that's actually good.

Responsibilities

• Build and iterate on prompt pipelines and multi-agent workflow components
• Design and integrate agentic workflows orchestrate multi-step, tool-using agents that plan, call models, and hand off between stages in production
• Deploy and serve open-source models set up inference endpoints, manage GPU compute, and optimize for latency and cost
• Write evals compare outputs against references, quantify quality, and feed results back into the pipeline
• Work on data pipelines: structured extraction from messy source text, localization, similarity/dedup
• Debug and maintain pipeline stages in production

What you bring:  

• (1+/3+) years of engineering experience, or a strong portfolio of shipped projects
• Solid Python fundamentals clean, working, readable code
• Hands-on experience with LLM APIs and prompt engineering (personal projects count)
• Comfort with Git, REST APIs, and working in a Linux environment
• A feel for content and narrative you can judge whether generated output is actually good, not just valid
• Curiosity and clear communication you ask good questions and don't stay stuck silently

 Preferred

• Exposure to agent/orchestration frameworks (LangGraph, LangChain, CrewAI)
• Familiarity with vector databases, embeddings, or RAG (Qdrant, pgvector)
• Hands-on work with open-source generative media models Flux, LTX, Wan, or similar
• Experience deploying open-source models for inference (vLLM, ComfyUI, Replicate/Cog, Docker + GPU)
• Experience writing evals or LLM-as-judge scoring
• Node.js and Fastapi familiarity, or experience deploying on AWS

About the role

We are seeking an AI Engineer to build and implement AI systems for content production at scale. You'll work at the intersection of engineering and content designing prompt pipelines, integrating generative models, and building the tooling that turns source material into finished creative output. The ideal candidate is technically strong but also has taste: someone who understands story and craft, and can tell the difference between output that's technically correct and output that's actually good.

Responsibilities

• Build and iterate on prompt pipelines and multi-agent workflow components
• Design and integrate agentic workflows orchestrate multi-step, tool-using agents that plan, call models, and hand off between stages in production
• Deploy and serve open-source models set up inference endpoints, manage GPU compute, and optimize for latency and cost
• Write evals compare outputs against references, quantify quality, and feed results back into the pipeline
• Work on data pipelines: structured extraction from messy source text, localization, similarity/dedup
• Debug and maintain pipeline stages in production

What you bring:  

• (1+/3+) years of engineering experience, or a strong portfolio of shipped projects
• Solid Python fundamentals clean, working, readable code
• Hands-on experience with LLM APIs and prompt engineering (personal projects count)
• Comfort with Git, REST APIs, and working in a Linux environment
• A feel for content and narrative you can judge whether generated output is actually good, not just valid
• Curiosity and clear communication you ask good questions and don't stay stuck silently

 Preferred

• Exposure to agent/orchestration frameworks (LangGraph, LangChain, CrewAI)
• Familiarity with vector databases, embeddings, or RAG (Qdrant, pgvector)
• Hands-on work with open-source generative media models Flux, LTX, Wan, or similar
• Experience deploying open-source models for inference (vLLM, ComfyUI, Replicate/Cog, Docker + GPU)
• Experience writing evals or LLM-as-judge scoring
• Node.js and Fastapi familiarity, or experience deploying on AWS

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

11

Preferred (Experience 1) – Experience with MLFlow, Kubeflow, Airflow, Prefect, Feature Stores, Model Registry, or MLOps/LLMOps frameworks.

12

Preferred (Experience 2) – Experience working with Vector Databases, Spark, PySpark, distributed ML pipelines, large-scale data processing, or real-time ML systems..

13

Preferred (Experience 3) – Familiarity with Docker, Kubernetes, Azure, AWS, GCP, cloud-native AI deployments, and scalable ML architecture.

14

Preferred (Company) – Candidates from AI-first startups, Fintech, Banking, Lending, Fraud Analytics, Risk Analytics, Product Companies, SaaS organizations, or data-driven technology companies

15

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

AI based systems design and development, entire pipeline from image/ video ingest, metadata ingest, processing, encoding, transmitting.

Implementation and testing of advanced computer vision algorithms.

Dataset search, preparation, annotation, training, testing, fine tuning of vision CNN models. Multimodal AI, LLMs, hardware deployment, explainability.

Detailed analysis of results. Documentation, version control, client support, upgrades.

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.

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

Strong AI Engineer / Machine Learning Engineer profiles.

2

Mandatory (Experience 1) – Must have minimum 5+ 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 30 Years

11

Preferred (Experience 1) – Experience with MLFlow, Kubeflow, Airflow, Prefect, Feature Stores, Model Registry, or MLOps/LLMOps frameworks.

12

Preferred (Experience 2) – Experience working with Vector Databases, Spark, PySpark, distributed ML pipelines, large-scale data processing, or real-time ML systems..

13

Preferred (Experience 3) – Familiarity with Docker, Kubernetes, Azure, AWS, GCP, cloud-native AI deployments, and scalable ML architecture.

14

Preferred (Company) – Candidates from AI-first startups, Fintech, Banking, Lending, Fraud Analytics, Risk Analytics, Product Companies, SaaS organizations, or data-driven technology companies

15

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

Key Responsibilities

• Design, build, and deploy machine learning and AI models that power Transient.AI's core products (research

automation, document intelligence, investor matching, and workflow orchestration).

• Work on applied NLP/LLM systems, including retrieval-augmented generation, structured extraction from

unstructured financial documents, and model evaluation pipelines.

• Partner closely with product and founding engineers to translate capital markets workflows into scalable AI

systems.

• Own model performance, reliability, and cost — from experimentation through production deployment.

• Build and maintain data pipelines, feature stores, and evaluation frameworks to support rapid iteration.

• Ensure systems meet the compliance, auditability, and security standards required in regulated financial

environments.

What We're Looking For

• 5+ years of experience building and deploying machine learning or AI systems in production.• Strong hands-on experience with Python and modern ML/AI frameworks (PyTorch, TensorFlow, Hugging Face,

LangChain, or equivalent).

• Experience with LLMs — fine-tuning, prompt engineering, RAG architectures, or agentic systems — is highly

valued.

• Solid grounding in data structures, distributed systems, and MLOps practices (model serving, monitoring,

versioning).

• Prior experience at a strong product company, high-growth startup, or a top-tier engineering background

• Comfort operating in an early-stage, high-ownership environment with limited process and high ambiguity.

• Exposure to fintech, capital markets, or other regulated industries is a plus, though not mandatory