Job Description

We are looking for an AI/ML Engineer with hands-on experience in Generative AI, Large Language Models (LLMs), NLP, and Retrieval-Augmented Generation (RAG) to design, build, and deploy enterprise-grade AI solutions. The ideal candidate will have strong foundations in Python-based ML development, vector databases, prompt engineering, LLM orchestration, and the Azure AI ecosystem.

Responsibilities

Generative AI & LLM Engineering
Build and fine-tune LLM-based applications using models such as GPT, LLaMA, Mistral, Phi, etc.
Develop custom pipelines for RAG, grounding LLMs on enterprise documents (PDFs, Word, PPT, HTML, images).
Design robust prompt engineering strategies—system prompts, few-shot examples, evaluation prompts.
Implement guardrails using safety filters, grounding validation, and hallucination detection.
NLP Model Development
Develop and optimize NLP models for text classification, entity extraction, summarization, Q&A, topic modelling, and semantic search.
Apply transformer-based architectures (BERT, RoBERTa, T5, LLaMA-based models).
Build multilingual NLP pipelines where required.
RAG Pipelines & Knowledge Engineering
Create scalable RAG architectures using:

Vector databases (Azure AI Search, Pinecone, Weaviate, FAISS).
Chunking, metadata tagging, hybrid search, and document embeddings.

Build connectors for ingestion pipelines (ETL) for structured/unstructured sources.
Evaluate and optimize retrieval quality (R@K, MRR, context relevancy scores).
Azure AI & MLOps
Use Azure AI Search, Azure OpenAI, Azure Machine Learning, Azure Functions, Data Lake, and DevOps pipelines to build end-to-end solutions.
Deploy LLM-based APIs, microservices, and inference endpoints using AKS/ACI.
Implement model evaluation frameworks, monitoring (logging, latency, cost), and lifecycle management.
Software Engineering & API Development
Build Python-based backend services for LLM inference, embeddings, and RAG operations.
Develop REST APIs and integrate with downstream applications.
(Nice-to-Have) Build simple UI front-ends using React for demos and internal tools.
Collaboration & Documentation
Work with domain SMEs, product teams, and engineering to convert business needs into AI use cases.
Document model architectures, design decisions, and best practices.

Qualifications

Required Skills & Qualifications

3–5 years of experience in AI/ML, with at least 1–2 years in Generative AI/LLMs/NLP.
Strong proficiency in Python, PyTorch, Hugging Face Transformers, LangChain/LlamaIndex.
Hands-on experience with RAG architectures, vector stores, and embedding models.
Strong understanding of transformer architecture, tokenization, embeddings, and LLM evaluation metrics.
Experience with Azure AI stack—Azure OpenAI, Cognitive Search/AI Search, Azure ML, Data Lake.
Familiarity with MLOps tools for tracking, deployment, CI/CD.
Experience working with unstructured data at scale—documents, text, OCR extractions.

Nice-to-Have Skills

Experience with React.js for building AI dashboards and internal apps.
Exposure to knowledge graphs, ontology/metadata design, and structured knowledge retrieval.
Experience deploying LLMs on edge or private environments (ONNX Runtime, quantized models).
Familiarity with evaluation frameworks such as Ragas, DeepEval, or custom LLM eval pipelines.

Education

Bachelor's/Master's degree in Computer Science, Engineering, Data Science, AI/ML, or equivalent fields.

BGV:

Employment with WSP India is subject to the successful completion of a background verification (BGV) check conducted by a third-party agency appointed by WSP India.
Candidates are advised to ensure that all information provided during the recruitment process — including documents uploaded — is accurate and complete, both to WSP India and its BGV partner.

About Us

WSP is one of the world's leading professional services consulting firms. We are dedicated to our local communities and propelled by international brainpower. We are technical experts and strategic advisors including engineers, technicians, scientists, architects, planners, surveyors and environmental specialists, as well as other design, program and construction management professionals. We design lasting solutions in the Transportation & Infrastructure, Property & Buildings, Earth & Environment, Power & Energy, Resources and Industry sectors, as well as offering strategic advisory services. Our talented people around the globe engineer projects that will help societies grow for lifetimes to come.

With approximately 4,000 talented people across 3 locations (Noida, Bengaluru & Mumbai offices) in India and more than 73,000 globally, in 550 offices across 40 countries, we engineer projects that will help societies grow for lifetimes to come.

At WSP we draw on the diverse skills and capabilities of our employees globally to compete for the most exciting and complex projects across the world and bring the same level of expertise to our local communities. We are proud to be an international collective of innovative thinkers who work on the most complex problems. Unied under one strong brand, we use our local expertise, international reach and global scale to prepare our cities and environments for the future, connect communities and help societies thrive in built and natural ecosystems. True to our guiding principles, our business is built on four cornerstones: Our People, Our Clients, Our Operational Excellence and Our Expertise.

www.wsp.com

We are

Passionate people doing purposeful and sustainable work that helps shape our communities and the future.
A collaborative team that thrives on challenges and unconventional thinking.
A network of experts channeling our curiosity into creating solutions for complex issues.

Inspired by diversity, driven by inclusion, we work with passion and purpose.

Working with Us

At WSP, you can access our global scale, contribute to landmark projects and connect with the brightest minds in your field to do the best work of your life. You can embrace your curiosity in a culture that celebrates new ideas and diverse perspectives. You can experience a world of opportunity and the chance to shape a career as unique as you.

Our Hybrid Working Module

With us, you can operate in a flexible, agile, yet structured work environment and follow a Hybrid Work Model.

Maximize collaboration.
Maintain product quality and cultural integrity.
Balance community, collaboration, opportunity, productivity, and efficiency.

Health, Safety and Wellbeing

Our people are our greatest asset, and we prioritize a safe work environment. Health, safety, and wellbeing are integral to our culture, with each of us accountable for fostering a safe workplace through our Making Health and Safety Personal initiative. Our Zero Harm Vision drives us to reduce risks through innovative solutions, earning recognition for our global health and safety practices with the prestigious RoSPA Health and Safety Awards for six consecutive years.

Inclusivity and Diversity

WSP India is dedicated to fostering a sustainable and inclusive work environment where our greatest strength - Our People -feel valued, respected, and supported. We ensure an unbiased approach in hiring, promotion, and performance evaluation, regardless of age, gender identity, race, religion, sexual orientation, marital status, physical ability, education, social status, or cultural background.

Imagine a better future for you and a better future for us all.

Join our close-knit community of over 73,300 talented global professionals dedicated to making a positive impact. Together, we can make a difference in communities both near and far.

With us, you can.

Apply today.

NOTICE TO THIRD PARTY AGENCIES:

WSP does not accept unsolicited resumes from recruiters, employment agencies, or other staffing services. Unsolicited resumes include any resume or hiring document sent to WSP in the absence of a signed Service Agreement where WSP has expressly requested recruitment/staffing services specific to the position at hand. Any unsolicited resumes, including those submitted to hiring managers or other business leaders, will become the property of WSP and WSP will have the right to hire that candidate without reservation – no fee or other compensation will be owed or paid to the recruiter, employment agency, or other staffing service.

AI/ML Engineer - Generative AI