AI Research Apprentice

Origin; Bengaluru; On-site; INR 800,000 - 1,100,000; Full time

Job summary

Origin, formerly 10xConstruction, is building general-purpose autonomous robots for US construction. We seek an AI Research Apprentice to prototype diffusion-based generative models and multimodal vision architectures that turn raw site footage into high-quality training data.

You will design auto-annotation pipelines and optimize models for on-robot deployment. You will work with perception and robotics teams to integrate research prototypes into live ROS 2 stacks, advancing on-device inference

Qualifications

• Strong foundation in deep learning, probabilistic modeling and computer vision (coursework or research projects).
• Hands-on experience with diffusion models (e.g., DDPM, Latent Diffusion) in PyTorch or JAX.
• Familiarity with multimodal transformers / VLMs (CLIP, BLIP, Flamingo, LLaVA, etc.) and contrastive pre-training objectives.
• Working knowledge of data-centric AI: active learning, self-training, pseudo-labeling and large-scale annotation pipelines.
• Solid coding skills in Python, PyTorch / Lightning, plus git-driven workflows; bonus for C++ and CUDA kernels.
• Bonus: experience with on-device inference (TensorRT, ONNX Runtime) & synthetic data tools (Isaac Sim).

Responsibilities

• Design and train diffusion-based generative models for Realistic, high-resolution synthetic data.
• Train and improve the segmentation and surface-quality models deployed on our robots in the field.
• Develop training objectives for Vision–Language–Action (VLA) models that link text work orders to pixel-level segmentation masks and robot actions.
• Architect large-scale auto-annotation pipelines that transform unlabeled images / point-clouds into high-quality labels with minimal human input.
• Benchmark model performance on accuracy, latency and memory for deployment on Jetson-class hardware; compress with distillation or LoRA.
• Collaborate with perception and robotics teams to integrate research prototypes into live ROS 2 stacks.

Skills

Deep learning

Probabilistic modeling

Computer vision

Diffusion models

PyTorch

JAX

Multimodal transformers

VLMs

Python

PyTorch Lightning

Git workflows

CUDA kernels

TensorRT

ONNX Runtime

Isaac Sim

Data-centric AI

Tools

CUDA

TensorRT

ONNX Runtime

Isaac Sim

ROS 2

Jetson

Job description

About Origin

Origin (previously 10xConstruction) is building general-purpose autonomous robots for US construction to tackle rising costs, safety risks, and labor shortages. Our modular, multi-trade platform combines purpose-built hardware with real-time site intelligence to navigate complex environments and execute tasks with precision. Trained in high-fidelity simulation and already deployed on live sites, our robots deliver 5x faster execution, 250%+ margin expansion, and significant cost savings. Join India’s most talent-dense robotics team consisting of individuals from IITs, Stanford, UCLA, etc.

About the role

As an AI Research Apprentice you'll push the frontiers of generative and multimodal learning that power our autonomous robots. You will prototype diffusion-based vision models, vision–language architectures (VLAs/VLMs) and automated data-annotation pipelines that turn raw site footage into training gold.

Key Responsibilities

• Design and train diffusion-based generative models for realistic, high-resolution synthetic data.
• Train and improve the segmentation and surface-quality models deployed on our robots in the field.
• Develop training objectives for Vision–Language–Action (VLA) models that link text work orders to pixel-level segmentation masks and robot actions.
• Architect large-scale auto-annotation pipelines that transform unlabeled images / point-clouds into high-quality labels with minimal human input.
• Benchmark model performance on accuracy, latency and memory for deployment on Jetson-class hardware; compress with distillation or LoRA.
• Collaborate with perception and robotics teams to integrate research prototypes into live ROS 2 stacks.

Qualifications & Skills

• Strong foundation in deep learning, probabilistic modeling and computer vision (coursework or research projects).
• Hands-on experience with diffusion models (e.g., DDPM, Latent Diffusion) in PyTorch or JAX.
• Familiarity with multimodal transformers / VLMs (CLIP, BLIP, Flamingo, LLaVA, etc.) and contrastive pre-training objectives.
• Working knowledge of data-centric AI: active learning, self-training, pseudo-labeling and large-scale annotation pipelines.
• Solid coding skills in Python, PyTorch / Lightning, plus git-driven workflows; bonus for C++ and CUDA kernels.
• Bonus: experience with on-device inference (TensorRT, ONNX Runtime) & synthetic data tools (Isaac Sim).

Preferred Experiences

• PyTorch or JAX
• C++
• CUDA kernels
• ONNX Runtime
• TensorRT
• Isaac Sim
• Latent Diffusion

• Gain experience in a dynamic startup environment at the forefront of robotics and AI innovation.
• Contribute to the development of technology that will revolutionize the construction industry.
• Work alongside a talented and passionate team committed to making a real-world impact.
• Learn from experienced professionals and gain valuable skills in robotics software engineering.
• Gain practical experience in a real-world engineering environment.
• Contribute to a project with the potential to make a significant impact on the construction industry.