AI/ML Engineer

Build and optimize LLM, RAG, and multi-agent workflows for production use cases. Design retrieval systems using vector search, reranking, and metadata filtering. Fine-tune, evaluate, and benchmark models to improve output quality and business impact. Deploy AI/ML services with strong standards for latency, scalability, and observability. Develop data, inference, and experimentation pipelines in partnership with product and engineering teams.

AI/ML Engineer with approx. 5 years of experience building production RAG platforms, multi-agent systems, and LLM retrieval pipelines using LangChain, LangGraph, GEMMA, and XGBoost across VR/XR * Languages, Databases: Python, SQL, PySpark, Spark, PostgreSQL, MongoDB

• Gen AI & ML: LangChain, LangGraph, LlamaIndex, RAG, Claude 3.5 Sonnet, Mistral-7B, GPT-4o, Hugging Face (PEFT, TRL), LoRA, QLoRA, SFT, DPO, RAGAS, Prompt Engineering, GEMMA, UIR, DRAMA, Laser KNN, A/B Testing, NLP, Strong Python and SQL skills with solid machine learning and NLP fundamentals. Experience with LangChain, LangGraph, LlamaIndex, prompt engineering, and retrieval optimization. Familiarity with vector databases and search tools such as Pinecone, OpenSearch, FAISS, or pgvector. Experience with Docker, Kubernetes, MLflow, CI/CD, and cloud platforms such as Azure. Preferred Qualifications Experience fine-tuning open-source models using LoRA, QLoRA, SFT, or DPO. Background in high-scale, low-latency AI systems and enterprise data platforms. Ability to work across research, engineering, and product teams to ship AI solutions into production.

The pay range for this role is $120k - $122k per annum including any bonuses or variable pay. Tech Mahindra also offers benefits like medical, vision, dental, life, disability insurance and paid time off (including holidays, parental leave, and sick leave, as required by law). Ask our recruiters for more details on our Benefits package. The exact offer terms will depend on the skill level, educational qualifications, experience, and location of the candidate.