Machine Learning Engineer

Job Title: AI/ML Engineer

Location:

Employment Type:

Experience:

Reporting To:

About Workwise

buy, build, and scale

Role Overview

As an AI/ML Engineer, you’ll build the domain-specific AI pipelines that power our core workflows. You’ll work closely with our AI Head and product/engineering teams to convert raw industrial data into smart, automated systems used in real operations.

What You’ll Do

• Build AI pipelines to convert unstructured documents (BOQs, tenders, specs) into structured datasets.
• Develop parsing, retrieval, and reasoning systems across multi-format industrial data.
• Fine-tune and deploy LLMs and vision-language models for production use cases.
• Design continuous learning loops and retraining workflows.
• Collaborate with engineering and product teams to scale prototypes into production-ready systems.

What We’re Looking For

• Strong Python and ML fundamentals (PyTorch/TensorFlow).
• Experience with LLM frameworks (LangChain, LlamaIndex, etc.).
• Familiarity with Document AI/OCR models (PaddleOCR, LayoutLM, Donut, etc.).
• Understanding of embeddings, semantic search, RAG, and entity linking.
• Clean coding, modular design, and Git/GitHub proficiency.
• Ability to build scalable, production-ready ML components.

Bonus Skills

• Experience with document understanding or information extraction systems.
• Built feedback-based evaluation or retraining pipelines.
• AWS experience (S3, Lambda, EC2, SageMaker) for model deployment.
• Open-source contributions or published ML projects.

What You’ll Achieve

• Month 1:

  Understand domain & data workflows and ship an internal prototype.

• Month 2:

  Deliver an end-to-end customer-facing demo solving a real use case.

• Month 3+:

  Own and improve a production-grade AI feature from ingestion to deployment.

Why We?

• Build real-world, high-impact AI for India’s industrial ecosystem.
• Work directly with top talent across AI, engineering, and product.
• Own meaningful projects early in your journey.
• Fast-growth environment with leadership opportunities.
• Competitive compensation and long-term career growth