AI/ML Engineer

About the role

Join our AI team to build production LLM systems, RAG pipelines, and agentic workflows for enterprise clients. You will work with the latest models (GPT-4, Claude, Llama) and ship AI products that solve real business problems.

Responsibilities

• Design and ship production-grade RAG pipelines with vector databases (Pinecone, pgvector, Weaviate)
• Build AI agents with tool-use, multi-step reasoning, and long-term memory
• Fine-tune and evaluate LLMs (Llama 3, Mistral, Phi) for domain-specific tasks
• Implement guardrails: output validation, hallucination detection, prompt injection defense
• Benchmark accuracy, cost, and latency; optimize for each
• Collaborate with clients to translate business needs into AI systems

Requirements

• 3+ years of ML/AI engineering experience with production deployments
• Deep Python skills + one or more of PyTorch, TensorFlow, or JAX
• Hands-on with LangChain, LlamaIndex, or similar orchestration frameworks
• Strong understanding of transformer architecture, attention, and LLM internals
• Experience with vector databases and embedding models
• Strong written communication, you will document models and decisions

Nice to have

• Published research or open-source AI projects on GitHub
• Experience with fine-tuning via LoRA/QLoRA on consumer GPUs
• Familiarity with MLOps tooling (Weights & Biases, MLflow, Kubeflow)
• Computer vision or speech/audio ML background

Benefits

• Competitive salary (INR 15 to 30 LPA), top of market for AI roles
• GPU credits for research and side projects
• Annual INR 75,000 learning & conference budget (NeurIPS, ICML, etc.)
• Remote-first with optional Indore office
• Health insurance for you and dependents
• Paid time off: 24 days + national holidays
• Dedicated research days (10% time) for exploratory work