Domain AI Researcher

Wipro Technologies; San Francisco (CA); On-site; USD 240,000 - 413,000; Full time

Benefits offered by this job

Medical benefits

Dental benefits

Paid time off

Job summary

Wipro Limited is seeking a highly technical Domain AI Researcher to design secure execution sandboxes and generate synthetic data for multi-step reasoning tasks.

You will collaborate with the Lead Researcher to transform sandboxes into benchmarks and co-author research papers, with focus on coding and system-level tasks. Strong ML, software architecture, and Docker experience are essential.

Qualifications

• Academic background in CS, systems eng or related with ML focus.
• Proficient in Python, C++, or Rust and software architecture.
• Experience with Docker, CI/CD, and code-parsing tools (AST/Tree-sitter/LSP).
• Hands-on with LLM orchestration frameworks (LangChain, AutoGen) and observability tools.

Responsibilities

• Build secure execution sandboxes for evaluating code and agentic actions.
• Create data pipelines for synthetic data and high-signal SFT/RLHF data.
• Define ground-truth through edge-case scenarios and AST-level checks.
• Collaborate to publish benchmarks and co-author research papers.

Skills

Python

C++

Rust

CI/CD

Code parsing

LangChain

AutoGen

LLM orchestration

Education

Degree in Computer Science

Degree in Systems Engineering

ML-focused coursework

Tools

Docker

CI/CD pipelines

AST/Tree-sitter/LSP

Job description

Wipro Limited (NYSE: WIT, BSE: 507685, NSE: WIPRO) is a leading technology services and consulting company focused on building innovative solutions that address clients’ most complex digital transformation needs. Leveraging our holistic portfolio of capabilities in consulting, design, engineering, and operations, we help clients realize their boldest ambitions and build future-ready, sustainable businesses. With over 230,000 employees and business partners across 65 countries, we deliver on the promise of helping our customers, colleagues, and communities thrive in an ever-changing world. For additional information, visit us at www.wipro.com.

Job Description

Role Overview

Static datasets are no longer sufficient to evaluate frontier models; they must interact, plan, and execute within dynamic environments. We are seeking a highly technical Domain AI Researcher to build the ground truth for agentic workflows. You will design secure execution sandboxes, generate complex synthetic data, and identify the exact thresholds where models fail in multi-step reasoning tasks.

Key Responsibilities

• Build Execution Sandboxes: Architect secure, containerized environments (Docker, gVisor) to safely evaluate LLM-generated code, shell commands, and agentic actions across multi-file repositories.
• Drive Data Synthesis: Build pipelines for rejection sampling, self-instruct, and synthetic data generation to create high-signal SFT and RLHF data specifically tailored to coding and system-level tasks.
• Create the Ground Truth: Move beyond standard unit tests to curate complex edge-case scenarios, design AST-level validation checks, and measure functional accuracy on multi-step frameworks.
• Collaborate & Publish: Partner with the Lead Researcher to transform execution sandboxes into publishable, open-source benchmarks (analogous to specialized SWE-bench frameworks) and co-author resulting research papers.

Candidate Profile

• Academic Background: Degree in Computer Science, Systems Engineering, or a related field with a heavy emphasis on machine learning and software architecture.
• Technical Expertise: Advanced programming proficiency (Python, C++, or Rust) coupled with deep knowledge of Docker, CI/CD pipelines, and code-parsing tools (AST, Tree-sitter, LSP).
• AI Frameworks: Hands-on experience with LLM orchestration frameworks (e.g., LangChain, AutoGen) and observability/tracing tools.
• Analytical Intuition: A strong investigative mindset tailored to finding edge cases—such as context window retrieval failures or infinite agent loops—and engineering the datasets required to resolve them.

Expected annual pay for this role ranges from$240,000.00 to$412,500.00. Based on the position, the role is also eligible for Wipro’s standard benefits including a full range of medical and dental benefits options, disability insurance, paid time off (inclusive of sick leave), other paid and unpaid leave options

Reinvent your world.We are building a modern Wipro. We are an end-to-end digital transformation partner with the boldest ambitions. To realize them, we need people inspired by reinvention. Of yourself, your career, and your skills. We want to see the constant evolution of our business and our industry. It has always been in our DNA - as the world around us changes, so do we. Join a business powered by purpose and a place that empowers you to design your own reinvention.

Applications from people with disabilities are explicitly welcome.

We are an Equal Opportunity Employer. All qualified applicants will receive consideration for employment without regard to race, color, caste, creed, religion, gender, marital status, age, ethnic and national origin, gender identity, gender expression, sexual orientation, political orientation, disability status, protected veteran status, or any other characteristic protected by law.

Wipro is committed to creating an accessible, supportive, and inclusive workplace. Reasonable accommodation will be provided to all applicants including persons with disabilities, throughout the recruitment and selection process. Accommodations must be communicated in advance of the application, where possible, and will be reviewed on an individual basis. Wipro provides equal opportunities to all and values diversity.