AI Research Engineer

BuildxPartners; Bengaluru; On-site; INR 1,200,000 - 1,500,000; Full time

Job summary

BuildxPartners in Bangalore is seeking an AI Research Engineer to advance computer vision, generative AI, and multimodal systems for autonomous robotics. The role blends research with production engineering, taking ideas from experimentation through deployment on real robot platforms.

You will develop diffusion-based generative models, VLMs, and data-centric AI solutions, scaling to millions of images and sensor data, while optimizing for edge devices and collaborating with CV, robotics, and

Qualifications

• 3 to 7 years in Deep Learning, AI Research, CV or related R&D.
• MS/PhD in CS, EE, Robotics or related fields is highly relevant.
• Strong hands-on with Diffusion Models: DDPM, LDM, ControlNet, Generative Modeling, Domain Adaptation.
• Strong experience with VLMs / Multimodal Transformers, such as CLIP, BLIP-2, LLaVA, Flamingo.
• Proven experience with Active Learning, Pseudo-Labeling, Self-Training, Weak Supervision, and automated annotation.
• Advanced Python + PyTorch skills; JAX is a plus.

Responsibilities

• Research and develop Diffusion-based Generative AI models for synthetic data generation, defect simulation, photorealistic environments, and domain adaptation.
• Design and train VLMs/VLAs connecting text instructions, visual data, CAD/spatial information, and sensor inputs for scene understanding and intelligent decision-making.
• Build scalable auto-annotation and data-centric AI pipelines using Active Learning, Self-Training, Pseudo-Labeling, Weak Supervision, and Synthetic Data.
• Develop AI workflows capable of handling millions of images, video frames, and point clouds with minimal manual annotation.
• Optimize models for edge deployment using INT8 quantization, LoRA, Knowledge Distillation, TensorRT, ONNX Runtime, CUDA, and C++.
• Work with NVIDIA Jetson-class edge hardware and integrate AI models with robotics systems.
• Collaborate with Computer Vision, Perception, Robotics, and Controls teams.
• Contribute to technical documentation, research publications, and mentor junior engineers/interns.

Skills

Deep Learning

AI Research

Computer Vision

Diffusion Models

VLMs

Multimodal Transformers

Python

PyTorch

Production Deployment

Edge AI

Education

MS/PhD in CS/EE/Robotics

Tools

JAX

PyTorch Lightning

DeepSpeed

Ray

TensorRT

ONNX Runtime

CUDA

Job description

Bangalore South, India | Posted on 19/08/2026

BuildxPartners is a global talent solutions firm delivering end-to-end recruitment and workforce solutions across industries and geographies.

→ BuildxAlpha – Executive & Leadership Search Focused on C-suite, board, and global executive hiring.

→ BuildxSigma – Comprehensive Talent Across Levels Covering junior, mid-level, and senior professionals.

→ BuildxGCC – Global Capability Center Solutions Specializing in Build, Operate, Transfer (BOT) model for Global Capability Centers, GCC supports companies in setting up, scaling, and transferring GCCs.

Job Description

Role Overview

We are looking for an AI Research Engineer to work on advanced Computer Vision, Generative AI, Vision-Language Models (VLMs), and Vision-Language-Action (VLA) systems for autonomous robotics.

The role combines AI research with production engineering taking ideas from research and experimentation through evaluation, optimization, and deployment on real-world robotic systems.

This is a research-focused AI role working on Computer Vision, Generative AI, and multimodal models for autonomous systems. The role involves developing Diffusion Models, VLMs, and data-centric AI solutions to improve machine perception and understanding, while taking research models through experimentation, optimization, and deployment on real-world edge systems.

Key Responsibilities

• Research and develop Diffusion-based Generative AI models for synthetic data generation, defect simulation, photorealistic environments, and domain adaptation.
  
• Design and train VLMs/VLAs connecting text instructions, visual data, CAD/spatial information, and sensor inputs for scene understanding and intelligent decision-making.
  
• Build scalable auto-annotation and data-centric AI pipelines using Active Learning, Self-Training, Pseudo-Labeling, Weak Supervision, and Synthetic Data.
  
• Develop AI workflows capable of handling millions of images, video frames, and point clouds with minimal manual annotation.
  
• Optimize models for edge deployment using INT8 quantization, LoRA, Knowledge Distillation, TensorRT, ONNX Runtime, CUDA, and C++ .
  
• Work with NVIDIA Jetson-class edge hardware and integrate AI models with robotics systems.
  
• Collaborate with Computer Vision, Perception, Robotics, and Controls teams.
  
• Contribute to technical documentation, research publications, and mentor junior engineers/interns.
  

Requirements

Must-Have Skills & Experience

• 3 to 7 years in Deep Learning, AI Research, Computer Vision, or related R&D; M.S./Ph.D. in CS, EE, Robotics, or related fields is highly relevant.
  
• Strong hands-on experience with Diffusion Models: DDPM, LDM, ControlNet, Generative Modeling, Synthetic Data, or Domain Adaptation.
  
• Strong experience with VLMs / Multimodal Transformers , such as CLIP, BLIP-2, LLaVA, Flamingo , or equivalent architectures.
  
• Proven experience with Active Learning, Pseudo-Labeling, Self-Training, Weak Supervision, and automated annotation .
  
• Advanced Python + PyTorch skills; JAX is a plus.
  
• Experience with scalable training frameworks such as PyTorch Lightning, DeepSpeed, Ray , or equivalent.
  
• Strong understanding of Probability, Optimization, Linear Algebra, and Information Theory .
  
• Ability to translate research concepts into production-ready AI systems .
  

Good to Have

• ROS 2, Isaac Sim, Open3D, Nav2, MoveIt 2
  
• 3D Computer Vision / Point Clouds
  
• TensorRT / ONNX Runtime / CUDA / C++
  
• Experience deploying models on NVIDIA Jetson or edge AI devices
  
• Strong research publications in AI/ML, Computer Vision, Generative AI, or Robotics.