AL/ML Engineer

Digiwagon
Ahmedabad, Gujarat

Apply Now

Job details

Full-time

Benefits

Paid time off

Qualifications

TensorFlow
Law
Software deployment
Master's degree
Analysis skills
Machine learning
Deep learning
Natural language processing
AI
Communication skills
Python

Full job description

Job Information

Date Opened

01/21/2026

Number of Positions

Industry

IT Services

Job Type

Full time

Work Experience

2-5 Years

City

Ahmedabad

State/Province

Gujarat

Country

India

Zip/Postal Code

380054

About Us

Digiwagon Technologies Pvt Ltd is a next-generation digital engineering and AI innovation company driving enterprise transformation through intelligent, scalable, and secure technology ecosystems.

We specialize in AI-powered applications, custom software engineering, eCommerce solutions, cloud-native architectures, mobile and web development, and advanced UI/UX systems that accelerate digital evolution and operational excellence.

By integrating artificial intelligence, automation, and human-centered design, Digiwagon empowers enterprises to enhance agility, deliver intelligent customer experiences, and achieve sustainable digital growth in a data-driven era.

Job Description

DigiWagon 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 LangChain, LangGraph, 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, TensorFlow, 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 LangChain and LangGraph
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 new 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: DigiWagon Technologies Pvt Ltd, 10th Floor, 1013 - Shivalik Shilp, ISKCON Cross Road, S.G. Highway, Ahmedabad, Gujarat 380054

Apply Now

Job Information

About Us

Job Description

Requirements

Benefits

Jobseeker tools

Employer Tools

Browse

Stay Connected