Senior AI Architect

About our company

We are a WhatsApp-based marketing and engagement platform that helps businesses grow their revenues via WhatsApp.

• We enable 100,000+ businesses with WhatsApp engagement and marketing.
• 400 Crores+ WhatsApp messages are exchanged between businesses and users per year.
• We work with top brands like Delhi Transport Corporation, Vivo, Physicswallah, and more.
• Biz owners drive 25-80% of their revenues using our platform.
• We are a mission-driven and growth-stage startup backed by investors.

Job Overview

We're looking for a Senior Machine Learning Engineer to lead the development and deployment of cutting-edge AI/ML systems with a strong focus on LLMs, Retrieval-Augmented Generation (RAG), AI agents, and intelligent automation.

You'll work closely with cross-functional teams to translate business needs into AI solutions, bringing your expertise in building scalable ML infrastructure, deploying models in production, and staying at the forefront of AI innovation.

Key Responsibilities

AI & ML System Development

• Design, develop, and optimize end-to-end ML models using LLMs, transformer architectures, and custom RAG frameworks.
• Fine-tune and evaluate generative and NLP models for business-specific applications such as conversational flows, auto-replies, and intelligent routing.
• Lead prompt engineering and build autonomous AI agents capable of executing multi-step reasoning.

Infrastructure, Deployment & MLOps

• Architect and automate scalable training, validation, and deployment pipelines using tools like MLflow, SageMaker, or Vertex AI.
• Integrate ML models with APIs, databases, and production services ensuring performance, reliability, and traceability.
• Monitor model performance in real-time, and implement A/B testing, drift detection, and re-training pipelines.

Data & Feature Engineering

• Work with structured, unstructured, and semi-structured data (text, embeddings, chat history).
• Build and manage vector databases and graph-based retrieval systems.
• Ensure high-quality data ingestion, feature pipelines, and scalable pre-processing workflows.

Team Collaboration & Technical Leadership

• Collaborate with product managers, software engineers, and stakeholders to align AI roadmaps with product goals.
• Mentor junior engineers and establish best practices in experimentation, reproducibility, and deployment.
• Stay updated on the latest in AI/ML, and drive innovation in applied use cases.

Required Skills and Qualifications

• Bachelor's/Master's degree in Computer Science, Engineering, AI/ML, or a related field from a reputable institution.
• 2+ years of experience building and deploying machine learning models in production.
• Expertise in Python and known frameworks like TensorFlow, PyTorch, Hugging Face, Scikit-learn.
• Hands-on experience with transformer models, LLMs, LangChain, LangGraph, OpenAI API, or similar.
• Deep knowledge of machine learning algorithms, model evaluation, hyperparameter tuning, and optimization.
• Experience working with cloud platforms and ML Ops tools.
• Strong understanding of SQL, data engineering concepts, and working with large-scale datasets.

Preferred Qualifications

• Experience with prompt tuning, agentic AI systems, or multi-modal learning.
• Familiarity with vector search systems and knowledge graphs.
• Contributions to open-source AI/ML projects or publications in AI journals/conferences.
• Experience in building conversational AI or smart assistants using messaging APIs.

Benefits

• Build AI that directly powers growth for 100,000+ businesses.
• Work on cutting-edge technologies like LLMs, RAG, and AI agents in real production environments.
• High ownership, fast iterations, and impact-focused work.

Others

• We are committed to delivering innovative solutions that meet the evolving needs of businesses.
• We foster a collaborative and dynamic work environment that encourages creativity and growth.