Lead AI Engineer

Lead AI Engineer

• *Location:** Bengaluru (On\-site) \| 5\-day work week
• *Compensation:** ₹40–60 LPA (Cash) \+ ₹0–20 LPA (ESOPs) — stretchable for exceptional candidates

• *About the Role**

We are looking for a Lead AI Engineer to drive the research and deployment of next\-generation perception and generative AI systems that power our autonomous construction robots. You will own the full lifecycle — from model architecture and large\-scale data pipelines to edge deployment on real hardware — working at the intersection of cutting\-edge AI research and production robotics.

• *Must\-Haves**

• 8\+ years in deep learning R\&D, or a PhD/MS in CS/EE/Robotics with a strong publication record
• Demonstrated expertise in diffusion models (DDPM, LDM, ControlNet) and Vision\-Language Models (CLIP, BLIP\-2, LLaVA)
• Proven experience building large\-scale data pipelines — active learning, pseudo\-labeling, synthetic data generation
• Advanced proficiency in Python, PyTorch/JAX, and scalable training frameworks (DeepSpeed, Ray, PyTorch Lightning)
• Edge AI deployment experience — TensorRT, ONNX Runtime, CUDA/C\+\+ optimization on Jetson\-class devices

• *What You'll Do**

• Lead research and development of diffusion and vision\-language models tailored to construction robotics use cases
• Architect and scale data pipelines for active learning, pseudo\-labeling, and synthetic data generation
• Own model training at scale using distributed frameworks and drive optimization for real\-time, on\-device inference
• Deploy and optimize models on edge hardware (Jetson\-class devices) for production robots operating on live sites
• Collaborate closely with robotics, perception, and hardware teams to ship AI capabilities into deployed systems
• Mentor and guide engineers, setting technical direction for the AI team

Pay: ₹4,000,000\.00 \- ₹6,000,000\.00 per year

Work Location: In person