Lead ML/NLP 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

Job Description:

We are seeking a versatile and highly skilled Lead AI/ML Engineer with deep expertise in Generative AI (GenAI) and Large Language Models (LLMs). This role requires a leader who can take full ownership of the

AI lifecycle—from initial architectural design to final production execution. You will lead the development of scalable AI-powered applications, demonstrating exceptional execution skills and the ability to deliver high-performance results under pressure in demanding production environments.

Machine Learning & LLM Capability:

 End-to-End ML Engineering: Build and manage comprehensive ML pipelines, including data ingestion, preprocessing, training, and evaluation using frameworks like PyTorch, TensorFlow, and Scikit-learn.  Advanced LLM Systems: Design and implement sophisticated LLM-based applications such as autonomous agents, chatbots, and complex automation tools.

 Generative AI Specialization: Architect and optimize Retrieval-Augmented Generation (RAG) pipelines using vector databases like FAISS, Pinecone, or Weaviate.

 Model Optimization: Fine-tune open-source and proprietary models (e.g., LLaMA, GPT) using advanced techniques like LoRA, QLoRA, or instruction tuning.

 Agentic Frameworks: Develop complex agentic workflows utilizing frameworks such as LangChain or LlamaIndex.

 Prompt Engineering: Implement expert-level prompt engineering, tool/function calling, and structured output generation.

 Project Ownership & Execution

 Full Lifecycle Ownership: Take complete accountability for the full ML and GenAI lifecycle, spanning data processing, model development, monitoring, and optimization.

 Architectural Leadership: Drive strategic architectural decisions for AI platforms, ensuring they are modular, scalable, and maintainable.

 Execution Excellence: Write clean, high-performance Python code following strict OOP principles and manage CI/CD pipelines for seamless project execution.

 Leadership & Mentoring: Act as a key technical leader, managing stakeholders and mentoring team members to ensure all project milestones are met with quality.

 System Integrity: Manage model and prompt versioning, experiment tracking, and comprehensive documentation for all pipelines and workflows.

 Performance Under Pressure

 Production Reliability: Ensure all AI systems maintain extreme scalability and performance under heavy production workloads, including both batch and real-time processing.

 High-Pressure Optimization: Rapidly optimize inference latency and system costs for ML and LLM systems to meet urgent business and technical requirements.

 Proactive Problem Solving: Apply strong analytical thinking to address complex challenges such as system drift, hallucinations, and latency in fast-paced environments.

 Robust Guardrails: Implement and manage strict evaluation frameworks and feedback loops to maintain system quality under stress.

Qualifications:

 Bachelor’s or Master’s degree in Computer Science, AI, ML, or a related field.

 Proven expertise in Python, system design, and scalable AI/ML architecture.

 Deep knowledge of NLP, Computer Vision, and Deep Learning models.

 Hands-on experience with Docker, Kubernetes, MLOps, and major cloud platforms (AWS, GCP, or Azure).

🔹 Key Responsibilities

• Design, develop, and deploy production-grade AI/ML and Generative AI solutions

• Work on GEO, AEO, and SGE initiatives to improve visibility across AI-driven search platforms

• Optimize content and digital experiences for conversational queries and LLM-based search

• Develop solutions using LLMs, NLP, embeddings, semantic search, RAG, and vector databases

• Analyze search intent, AI-generated responses, citations, retrieval patterns, and content discoverability

• Build frameworks to measure GEO/AEO strategies and AI-search performance

• Collaborate with Product, Engineering, Content, SEO, Marketing, and Business teams

• Improve solution accuracy, relevance, latency, and user experience

🔹 Mandatory Requirements

✅ 1–4 years of professional experience

✅ Minimum 1 year of hands-on experience in GEO, AEO, or SGE

✅ Experience with prompt engineering, embeddings, vector search, or RAG systems

✅ Understanding of semantic search and entity-based optimization

✅ Exposure to ChatGPT, Google Gemini, or similar LLM platforms

✅ Knowledge of schema, context building, content structuring, and knowledge representation

🎓 Preferred Education

B.Tech, M.Tech, Integrated M.Sc., or MS from a Tier-1 engineering institute such as IIT, NIT, BITS, VIT, DTU, or NSUT.