Lead AI Engineer

Posted 7 Days Ago
Be an Early Applicant
Bengaluru, Bengaluru Urban, Karnataka, IND
In-Office
Senior level
Artificial Intelligence • HR Tech • Professional Services • Software
The Role
Lead R&D on diffusion-based generative models and vision-language systems for photorealistic simulation, defect synthesis, and domain adaptation. Build large-scale auto-annotation and synthetic-data pipelines, optimize and compress models for Jetson-class edge devices under ROS2, oversee full project lifecycle, publish research, and mentor junior engineers.
Summary Generated by Built In

This role is for one of the Weekday's clients

Min Experience: 8+ years

Location: Bengaluru
JobType: full-time


Requirements

Key Responsibilities

  • Research & innovate diffusion-based generative models for photorealistic wall-surface simulation, defect synthesis and domain adaptation.
  • Architect and train Vision-Language Models (VLMs) and Vision-Language Action Models (VLA) objectives that connect textual work orders, CAD plans and sensor data to pixel-level understanding.
  • Lead development of auto-annotation pipelines (active learning, self-training, synthetic data) that scale to millions of frames and point-clouds with minimal human effort.
  • Optimize and compress models (INT8, LoRA, distillation) for deployment on Jetson-class edge devices under ROS 2.
  • Own the full lifecycle—problem definition, literature review, prototyping, offline/online evaluation and production hand-off to perception & controls teams.
  • Publish internal tech reports and external conference papers; mentor interns and junior engineers.
Requirements

Qualifications & Skills

  • 8+ years in deep-learning R&D or Ph.D./M.S. in CS, EE, Robotics or related field with strong publication record.
  • Demonstrated expertise in diffusion models (DDPM, LDM, ControlNet) and multimodal transformers / VLMs (CLIP, BLIP-2, LLaVA, Flamingo).
  • Proven success building large-scale data-centric AI workflows—active learning, pseudo-labeling, weak supervision.
  • Advanced proficiency in Python, PyTorch (or JAX), experiment tracking and scalable training (PyTorch Lightning, DeepSpeed, Ray).
  • Familiarity with edge-AI runtimes (TensorRT, ONNX Runtime), and CUDA / C++ performance tuning.
  • Strong mathematical foundation (probability, information theory, optimization) and ability to translate theory into production code.
  • Bonus: experience with synthetic data generation in Isaac Sim or robotics perception stacks (ROS2, Nav2, MoveIt 2, Open3D).
Must-have skills

AI, Diffusion Models, Python

Good-to-have skills

DDPM, CLIP, Pytorch

Skills Required

  • 8+ years in deep-learning R&D or M.S./Ph.D. in CS, EE, Robotics with strong publication record
  • Demonstrated expertise in diffusion models (DDPM, LDM, ControlNet)
  • Expertise in multimodal transformers / VLMs (CLIP, BLIP-2, LLaVA, Flamingo)
  • Proven experience building large-scale data-centric AI workflows (active learning, pseudo-labeling, weak supervision)
  • Advanced proficiency in Python and PyTorch (or JAX)
  • Experience with scalable training and experiment tooling (PyTorch Lightning, DeepSpeed, Ray)
  • Familiarity with edge-AI runtimes and deployment (TensorRT, ONNX Runtime) and CUDA/C++ performance tuning
  • Strong mathematical foundation (probability, information theory, optimization)
  • Experience optimizing/compressing models (INT8 quantization, LoRA, distillation) for Jetson-class edge devices
  • Experience with synthetic data generation in Isaac Sim or robotics perception stacks (ROS2, Nav2, MoveIt 2, Open3D)
  • Publication record and ability to publish external conference papers; mentoring interns/junior engineers
Am I A Good Fit?
beta
Get Personalized Job Insights.
Our AI-powered fit analysis compares your resume with a job listing so you know if your skills & experience align.

The Company
Year Founded: 2021

What We Do

Weekday is an AI-powered recruitment platform that helps startups hire top-tier engineering and product talent. By leveraging a massive database of white-collar professionals and advanced outreach tools, the company streamlines the hiring process through automated sourcing, AI-driven resume screening, and white-glove contingency services. Their mission is to modernize recruitment by enabling companies to discover and engage passive candidates efficiently, ensuring high-quality hires for critical roles.

Similar Jobs

Coupa Logo Coupa

Artificial Intelligence Engineer

Artificial Intelligence • Fintech • Information Technology • Logistics • Payments • Business Intelligence • Generative AI
Hybrid
Bangalore, Bengaluru Urban, Karnataka, IND
3000 Employees

Optum Logo Optum

Site Reliability Engineer

Artificial Intelligence • Big Data • Healthtech • Information Technology • Machine Learning • Software • Analytics
In-Office
Bengaluru, Bengaluru Urban, Karnataka, IND
160000 Employees

Optum Logo Optum

Lead Data Engineer

Artificial Intelligence • Big Data • Healthtech • Information Technology • Machine Learning • Software • Analytics
In-Office
Bengaluru, Bengaluru Urban, Karnataka, IND
160000 Employees
In-Office
Bangalore, Bengaluru Urban, Karnataka, IND
29154 Employees

Similar Companies Hiring

Hanover Park Thumbnail
Artificial Intelligence • Fintech • Software • Financial Services
New York, New York
42 Employees
Kepler  Thumbnail
Fintech • Software
New York, New York
6 Employees
Onshore Thumbnail
Artificial Intelligence • Fintech • Software • Financial Services
New York, New York
60 Employees

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account