AI/ML Engineer – LLM & Python

We are looking for a skilled AI/ML Engineer – LLM & Python to join our growing AI team in Mohali. The ideal candidate should have strong Python development skills and hands-on experience building AI-powered applications using Large Language Models (LLMs), Generative AI, RAG, and modern AI frameworks.

You will work on real-world AI solutions involving intelligent assistants, document processing, automation, AI agents, and LLM-powered applications. You will collaborate with developers, project managers, and product teams to design, develop, integrate, and optimize scalable AI solutions for clients across different industries.

Key Responsibilities

• Design, develop, and deploy AI-powered applications using Python and Generative AI technologies.
• Build applications using LLMs such as OpenAI GPT, Anthropic Claude, Google Gemini, Llama, Qwen, and other open-source models.
• Develop and optimize Retrieval-Augmented Generation (RAG) pipelines for AI applications.
• Work with embeddings, semantic search, document processing, and vector databases.
• Build intelligent AI workflows and agents using frameworks such as LangChain, LangGraph, CrewAI, AutoGen, or similar technologies.
• Develop and integrate REST APIs and backend services for AI-powered applications.
• Integrate LLM APIs and third-party AI services into existing applications.
• Design effective prompts and improve response quality, accuracy, consistency, and reliability.
• Evaluate AI outputs and identify opportunities to reduce hallucinations and improve overall system performance.
• Work with structured and unstructured data to prepare information for AI and LLM applications.
• Optimize AI applications for latency, scalability, reliability, and cost.
• Debug, troubleshoot, and resolve issues across AI pipelines and backend services.
• Collaborate with software developers, product teams, and other stakeholders to translate business requirements into practical AI solutions.
• Document AI workflows, integrations, technical decisions, and development processes.
• Stay updated with advancements in Generative AI, LLMs, AI agents, NLP, and AI development frameworks.

Required Skills & Experience

• 1–4 years of professional experience in AI/ML Engineering, Machine Learning, Python Development, or a related field.
• Strong proficiency in Python.
• Hands-on experience working with Generative AI and Large Language Models (LLMs).
• Good understanding of Machine Learning, Deep Learning, and NLP fundamentals.
• Practical experience with Prompt Engineering.
• Hands-on experience building or integrating RAG applications.
• Understanding of embeddings, semantic search, and vector databases.
• Experience with at least one LLM/AI framework such as LangChain, LangGraph, LlamaIndex, CrewAI, or similar.
• Experience integrating AI APIs such as OpenAI, Anthropic, Google Gemini, or Hugging Face.
• Good understanding of REST APIs and backend development.
• Familiarity with databases such as MySQL, PostgreSQL, MongoDB, or similar.
• Familiarity with Git and version control systems.
• Strong analytical, debugging, and problem-solving skills.
• Ability to work independently and collaborate effectively with a cross-functional team.

Good to Have

• Experience building AI Agents or Agentic AI workflows.
• Knowledge of MCP (Model Context Protocol).
• Experience with vector databases such as Pinecone, ChromaDB, FAISS, Weaviate, or Qdrant.
• Familiarity with Hugging Face Transformers and open-source LLMs.
• Experience with PyTorch or TensorFlow.
• Knowledge of LLM evaluation, guardrails, structured outputs, or function/tool calling.
• Experience with FastAPI, Flask, or Django.
• Familiarity with Docker and containerized application deployment.
• Exposure to AWS, Azure, or Google Cloud.
• Basic understanding of MLOps/LLMOps and production AI deployment.
• Experience with workflow automation platforms such as n8n, Make, or Zapier.
• GitHub projects, AI/ML portfolio, or experience building AI prototypes and production applications.

Why Join Us?

• Competitive salary based on skills and experience.
• Opportunity to work on real-world Generative AI and LLM projects.
• Exposure to the latest AI models, frameworks, agentic technologies, and automation tools.
• Opportunity to work on international client projects across different industries.
• Collaborative and innovation-driven work environment.
• Career growth opportunities in the rapidly evolving AI/ML domain.
• Continuous learning and professional development opportunities.
• Flexible working environment.

Pay: ₹15,000.00 - ₹40,000.00 per month

Benefits

• Flexible schedule
• Leave encashment
• Life insurance
• Paid sick time
• Paid time off
• Provident Fund

Work Location: In person