AI/ML Engineer

Job Role: AI/ML Engineer

• We are seeking an AI Engineer to design and deploy production-grade AI solutions across LLM applications, RAG pipelines, retrieval systems, and scalable ML services. This role blends model orchestration, search relevance, MLOps, and cross-functional collaboration to deliver reliable, high-impact AI products.
• 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

Responsibilities:-

• 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.

Required Qualifications

• 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.

Tech Mahindra is an Equal Employment Opportunity employer. We promote and support a diverse workforce at all levels of the company. All qualified applicants will receive consideration for employment without regard to race, religion, color, sex, age, national origin or disability. All applicants will be evaluated solely on the basis of their ability, competence, and performance of the essential functions of their positions with or without reasonable accommodations. Reasonable accommodations also are available in the hiring process for applicants with disabilities.