AI Engineer

We are seeking a hands-on AI/ML Engineer with deep expertise in RAG (Retrieval-Augmented Generation) agents, Small Language Model (SLM) fine-tuning, and custom dataset workflows. You'll work closely with our AI research and product teams to build production-grade models, deploy APIs, and enable next-gen AI-powered experiences.

Responsibilities:

• Design and build RAG-based solutions using vector databases and semantic search.
• Fine-tune open-source SLMs (e. g., Mistral, LLaMA, Phi, etc. ) on custom datasets.
• Develop robust training and evaluation pipelines with reproducibility.
• Create and expose REST APIs for model inference using FastAPI.
• Build lightweight frontends or internal demos with Streamlit for rapid validation.
• Analyze model performance and iterate quickly on experiments.
• Document processes and contribute to knowledge sharing within the team.

Requirements:

• 3-5 years of experience in applied ML/AI engineering roles.
• Expert in Python and common AI frameworks (Transformers, PyTorch/TensorFlow).
• Deep understanding of RAG architecture and vector stores (FAISS, Pinecone, Weaviate).
• Experience with fine-tuning transformer models and instruction-tuned SLMs.
• Proficient with FastAPI for backend API deployment and Streamlit for prototyping.
• Knowledge of tokenization, embeddings, training loops, and evaluation metrics.

Nice to Have:

• Familiarity with LangChain, the Hugging Face ecosystem, and OpenAI APIs.
• Experience with Docker, GitHub Actions, and cloud model deployment (AWS/GCP/Azure).
• Exposure to experiment tracking tools like MLFlow and Weights and Biases.