Digi Wagon builds AI-powered data-driven products that help enterprises improve agility automate intelligence and deliver meaningful customer experiences By combining artificial intelligence automation and human-centered design we enable organizations to scale innovation and achieve sustainable digital growth We are looking for an AI ML Engineer with hands-on experience in building and deploying LLM-powered applications RAG pipelines and AI-driven systems You will work on real-world AI ML products involving NLP embeddings vector search and agent-based workflows collaborating closely with backend data and product teams to deliver production-ready AI solutions This role is ideal for engineers who enjoy applied AI system design and transforming ML concepts into scalable usable applications Requirements Strong proficiency in Python for AI ML development Experience working on AI ML-based products particularly LLM-powered applications Hands-on experience with Lang Chain Lang Graph and prompt engineering techniques Strong understanding of prompt design prompt optimization and RAG pipelines Experience designing AI ML application architectures involving LLMs Embeddings Vector databases Embedding search frameworks FAISS or similar Solid knowledge and hands-on experience in Natural Language Processing NLP Understanding of different ML model types and their use cases including Linear and logistic regression Neural networks Traditional ML vs deep learning approaches Experience with AI ML libraries and tools such as PyTorch Tensor Flow scikit-learn Hugging Face Experience or exposure to fine-tuning pre-trained models LLMs or traditional ML models for domain-specific use cases Knowledge and exposure to Agentic AI concepts including multi-step reasoning and tool-based agents Understanding of MLOps practices including model versioning experimentation deployment and monitoring Experience validating and testing AI ML pipelines models and outputs Ability to integrate existing ML models into production pipelines and applications Strong analytical thinking problem-solving skills and ability to collaborate with cross-functional teams Good communication skills and a product-oriented mindset Responsibilities Design and develop AI ML-powered applications using LLMs embeddings and vector search Build optimize and maintain RAG pipelines for retrieval-based AI systems Implement effective prompt engineering and prompt optimization strategies Develop and integrate NLP solutions for real-world use cases Fine-tune pre-trained models when required to improve domain relevance accuracy or performance Design and integrate Agentic AI workflows including Tool-based agents multi-step reasoning pipelines Orchestration using frameworks like Lang Chain and Lang Graph Select evaluate and apply appropriate ML models based on product requirements Integrate custom and pre-trained ML models into scalable AI pipelines Collaborate with backend and platform teams to deploy AI services into production Apply MLOps best practices for model lifecycle management and monitoring Test evaluate and validate AI ML systems for quality reliability and performance Troubleshoot AI ML pipeline issues and perform Root Cause Analysis RCA Continuously explore and adopt current AI ML techniques to enhance product capabilities Benefits 5 Days Working with half-day on 1st Saturday of each month 15 Paid leaves a year 3 additional Experience Based Leave 12 Holiday Leaves A Year Team Building Activities Annual Company Trip Training And Professional Development Employee Friendly Culture Celebrations And Social Events Location Digi Wagon Technologies Pvt Ltd 10th Floor 1013 - Shivalik Shilp ISKCON Cross Road S G Highway Ahmedabad Gujarat 380054 Follow us for more details on

📌 Al/ml Engineer (Vadodara)
🏢 Digiwagon Technologies Private
📍 Vadodara