Sr / Lead AI Engineer

Wishtreetech; Ahmedabad District; On-site; INR 1,200,000 - 1,800,000; Full time

Job summary

Wishtreetech is seeking an experienced AI Engineer to design and implement end-to-end AI/ML architectures for enterprise-scale Generative AI products in Ahmedabad, India. The role requires proficiency in Python and various ML frameworks, including deep experience in deploying large language models.

The ideal candidate will lead projects, mentoring junior engineers and collaborating with cross-functional teams to develop AI solutions meeting business needs. A firm understanding of data privacy regulations and cloud platforms such as AWS and Azure is essential.

Qualifications

• 2+ years of hands-on experience in Generative AI and Machine Learning in a production environment.
• Proficiency in Python and ML frameworks like PyTorch and TensorFlow.
• Strong knowledge of RAG architectures and embedding models.

Responsibilities

• Design and architect end-to-end AI/ML systems for Generative AI products.
• Lead deployment of large language models (LLMs) such as GPT-4 and Claude.
• Ensure compliance with data privacy regulations and internal security policies.

Skills

Generative AI

Machine Learning

Python

MLOps

NLP

Education

Bachelor's or Master's degree in Computer Science or related field

Tools

PyTorch

TensorFlow

Hugging Face Transformers

Docker

Kubernetes

AWS

Azure

GCP

Job description

• Overall 8+ years of experience in Design end-to-end AI/ML architectures for enterprise-scale Generative AI products, including LLM-based applications, RAG pipelines, and agentic workflows.
• Hands on Conversational AI agents, RAG(AS), NLP
• Lead the selection, fine-tuning, and deployment of large language models (LLMs) such as GPT-4, Claude, Llama, and Mistral.
• Architect robust data pipelines and vector databases (Pinecone, Weaviate, pgvector, Qdrant) to support retrieval-augmented generation (RAG) and embedding-based search.
• Define AI governance, responsible AI principles, and model observability frameworks across the organization.
• Collaborate with sales, engineering, and data science teams to translate business requirements into AI solutions.
• Evaluate and integrate third-party AI APIs, MLOps platforms, and cloud AI services (AWS SageMaker, Azure OpenAI, GCP Vertex AI).
• Mentor junior AI engineers and data scientists; champion best practices in model development and deployment.
• Stay current with research in foundation models, multi-modal AI, and emerging GenAI techniques, and translate insights into actionable product improvements.
• Lead proof-of-concept (PoC) projects and drive them to production-grade deployments with a focus on latency, cost, and reliability.
• Ensure AI systems comply with data privacy regulations (GDPR, DPDP) and internal security policies.

Required Qualifications

• 2+ years of hands-on experience in Generative AI and Machine Learning in a production environment.
• Proficiency in Python and ML frameworks: PyTorch, TensorFlow, Hugging Face Transformers.
• Proven experience building and deploying LLM-based applications (prompt engineering, fine-tuning, RLHF).
• Strong knowledge of RAG architectures, embedding models, and vector databases.
• Experience with cloud platforms (AWS, Azure, or GCP) and containerisation (Docker, Kubernetes).
• Solid understanding of MLOps practices: CI/CD for ML, model versioning, monitoring, and drift detection.
• Familiarity with orchestration frameworks such as LangChain, LlamaIndex, or AutoGen.
• Strong system design skills with the ability to architect scalable, fault-tolerant AI systems.
• Excellent communication skills to present technical concepts to non-technical stakeholders.
• Bachelor's or Master's degree in Computer Science, AI/ML, Data Science, or a related field (or equivalent practical experience).