Staff Research Engineer, Data Agents

United States Digital Space LLC; San Francisco (CA); On-site; USD 190,000 - 270,000; Full time

Benefits offered by this job

Health insurance

Paid parental leave

401(k) with company match

Paid time off

Job summary

United States Digital Space LLC, based in San Francisco, is seeking a skilled researcher to develop innovative Data agents that provide value to users. You'll work closely with product teams to refine prototypes into effective solutions and push the boundaries of enterprise AI.

Ideal candidates have 2+ years of applied research experience, a strong foundation in computer science, and a passion for enhancing user experiences with cutting-edge technology. Join us to shape the future of enterprise AI!

Qualifications

• 2+ years of experience in an applied research environment, shipping research prototypes to production.
• Ability to work effectively in a fast-moving environment blending research and product rigor.

Responsibilities

• Develop post-training recipes to train enterprise Data agents.
• Build systems that help agents discover and use relevant lakehouse context.
• Partner with product teams to enhance agentic experiences.

Skills

Experience in LLMs

Reinforcement learning

Applied research

Cross-functional collaboration

Education

BS, MS, or PhD in Computer Science or related field

Job description

About the company

The company is the Data + AI company. More than 10,000 organizations worldwide — including Comcast, Condé Nast, Grammarly, and over 60% of the Fortune 500 — rely on the company Lakehouse Platform to unify their data, analytics, and AI. The company is headquartered in San Francisco, with offices around the globe.

About the Team

The company AI Research organization is pushing the frontier of next‑generation enterprise AI. We believe a company’s data is its greatest competitive advantage, and we’re building the models and agents that unlock it. Our work spans the full stack, from model training to advanced multi‑agent systems.

The Data Agent team within AI Research is dedicated to pioneering frontier enterprise data agents. Our research focuses on developing agents capable of autonomous planning, code generation, and multi‑step workflow execution within intricate enterprise settings. The team’s primary technical pillars include post‑training enhancements, harness design, agentic reinforcement learning (RL), and the construction of specialized RL environments. As a member of this team, you will bridge the gap between cutting‑edge AI research and product development by shipping direct improvements to Genie, the company’s agent product that provides significant value to our user base.

Impact you will have

• Develop the best post‑training recipes to train enterprise Data agents, capable of autonomous planning, code generation, and multi‑step workflow execution within intricate enterprise settings.
• Partner closely with product teams to turn prototypes and research ideas into the best agentic experience for company users.
• Build systems that help the agent discover and use relevant lakehouse context, including tables, notebooks, code, and cell outputs, to produce more accurate and useful results.
• Raise the technical bar for the team through strong design, execution, debugging, and mentorship, helping shape the long‑term direction of agentic experiences at the company.

What we look for

• BS, MS, or PhD in Computer Science or a related field.
• 2+ years of experience in an applied research environment, with a track record of shipping research prototypes to production.
• Experience in LLMs, agents, reinforcement learning, post‑training workflows.
• Ability to work effectively in a fast‑moving environment that blends research exploration with product and engineering rigor.
• Clear communication and strong cross‑functional collaboration with researchers, engineers, and product stakeholders.

Pay Range Transparency

The company is committed to fair and equitable compensation practices. The pay range for this role is $190,000 – $270,000 USD. Total compensation may also include eligibility for an annual performance bonus, equity, and the benefits listed below.

Benefits

At the company, we strive to provide comprehensive benefits and perks that meet the needs of all employees. Benefits include health, dental, vision, life insurance, 401(k) with company match, paid parental leave, paid time off, and other perks. Specific details for your region are available upon request.

Our Commitment to Diversity and Inclusion

We are committed to fostering a diverse and inclusive culture where everyone can excel. We take great care to ensure that our hiring practices are inclusive and meet equal employment opportunity standards. Individuals looking for employment at the company are considered without regard to age, color, disability, ethnicity, family or marital status, gender identity or expression, language, national origin, physical and mental ability, political affiliation, race, religion, sexual orientation, socio‑economic status, veteran status, and other protected characteristics.

Compliance

If access to export‑controlled technology or source code is required for performance of job duties, it is within Employer's discretion whether to apply for a U.S. government license for such positions, and Employer may decline to proceed with an applicant on this basis alone.