AI Engineer - Llama / LangChain / RAG

Overview

We are seeking a skilled AI Engineer with 2-4 years of experience to design, develop, and deploy AI/ML solutions, with a strong emphasis on Generative AI, NLP and Retrieval-Augmented Generation (RAG). The ideal candidate will have hands-on experience with Llama, LangChain or LangGraph and other Generative AI frameworks to build cutting-edge AI applications.

Location: Mumbai City (Work from Office only)

Key Responsibilities:

• Develop and deploy AI/ML models with a focus on NLP, Generative AI, and RAG
• Engineer and refine prompts to optimize performance for large language models (LLMs) and generative AI applications.
• Implement and maintain data preprocessing pipelines, including data cleansing, feature engineering, embedding generation, and transformation for time series data.
• Utilize LangChain or LangGraph and other Generative AI frameworks to build structured workflows for LLM-based applications.
• Train and fine-tune machine learning models, including LLMs, to optimize performance for various use cases.
• Analyze and interpret complex datasets, including time series data, ensuring scalable and efficient AI model deployment.
• Collaborate with cross-functional teams to integrate AI/ML models into existing systems and workflows.
• Ensure models are robust, scalable, and adhere to best practices in ethical AI development.
• Conduct testing, performance benchmarking, and iterative refinements of AI models and pipelines.
• Stay updated with the latest advancements in AI/ML, NLP, and cloud technologies and recommend integration strategies for emerging tools and techniques.

Key Skills and Qualifications:

• AI/ML Expertise: Proven experience in developing, fine-tuning, and deploying AI/ML models, focusing on NLP, Generative AI, and RAG.
• Prompt Engineering: Proficiency in designing, testing, and optimizing prompts for LLMs.
• Data Analysis: Strong ability to process and interpret large and complex datasets, including time series data, for AI model training and validation.
• Programming: Proficiency in Python and experience with AI/ML frameworks like TensorFlow, PyTorch, and scikit-learn and experience with backend integration (e.g., FastAPI, Node.js)
• NLP Techniques: Understanding of tokenization, embeddings, transformer-based models (e.g., Llama), and RAG workflows.
• LangGraph & GenAI Frameworks: Hands-on experience with LangGraph, LangChain, HuggingFace Transformers, OpenAI API, and other Generative AI tools.
• Cloud Deployment: Familiarity with deploying AI solutions on AWS, GCP, or Azure.
• Model Training & Fine-Tuning: Ability to train and fine-tune machine learning models, including large language models (LLMs), to enhance performance and accuracy.

Job Type: Full-time

Pay: ₹1,200,000.00 - ₹2,000,000.00 per year

Ability to commute/relocate:

• Mumbai, Maharashtra: Reliably commute or planning to relocate before starting work (Required)

Application Question(s):

• Post selection, can you join immediately or within 30 days?

Experience:

• AI Engineer: 2 years (Required)

Work Location: In person