# Research Engineer - AI/RL Infrastructure

**Company:** [Applied Intuition](https://hotfix.jobs/companies/applied-intuition)
**Location:** Sunnyvale, CA
**Role:** ML Engineering
**Salary:** $126k – $423k/yr
**Skills:** PyTorch, CUDA, Ray, Flyte, Kubernetes, Gpu Clusters, Distributed Training, ML Infrastructure, Data Pipelines, Performance Engineering
**Posted:** 2026-04-21

> Designs, builds, and operates large-scale ML infrastructure for AI/RL research, including GPU cluster orchestration, data curation pipelines, and distributed training systems for autonomous driving and robotics.

## Job Description

## Responsibilities
- Design and build training and evaluation infrastructure to support AI research directions, orchestrating massive GPU clusters to process PBs of multimodal sensor data
- Build robust benchmarking, continuous evaluation, and regression tracking systems to measure model performance across diverse, long-tail real-world driving distributions
- Develop large-scale data sampling, dataset generation, and advanced data curation pipelines, leveraging state-of-the-art AI models to power a closed-loop data flywheel
- Enable high-throughput distributed training across heterogeneous cloud environments, focusing on reliability, efficiency, and cost-aware scaling
- Collaborate closely with AI research, autonomy, and platform teams to translate cutting-edge research into production-ready systems

## Requirements
- Experience building and operating production-grade software systems across the full machine learning lifecycle, including training, evaluation, data, and deployment
- Opinions about building a company-wide platform for ML training, evaluation, and deployment
- Experience with performance engineering and compute acceleration for large-scale ML training, including profiling, bottleneck analysis, and optimization
- Strong systems-level debugging skills to diagnose and resolve issues in large-scale distributed training, spanning model code, data pipelines, runtimes, and cluster infrastructure
- Deep familiarity with the open-source ML and systems ecosystem, with judgment on when to adopt open source versus build in-house
- Technical experience in: **PyTorch**, **CUDA**, **Ray**, **Flyte**, **Kubernetes**

## Nice to Have
- Industry experience on relevant topics (self-driving application preferred)

## Compensation
Base salary range: **$126,000 - $423,000 USD** annually, plus equity and benefits.

## Similar roles

- [Applied AI Engineer](https://hotfix.jobs/jobs/31682948-1fdf-4b99-9032-69a3fe90514c) - Snowflake - Menlo Park, CA - $126k – $182k/yr
- [Research Scientist - Reinforcement Learning, Self-Driving](https://hotfix.jobs/jobs/c0af727c-1cd0-471d-9452-26d2c1703aba) - Applied Intuition - Sunnyvale, CA - $126k – $423k/yr
- [Research Scientist - Reinforcement Learning, Robotics](https://hotfix.jobs/jobs/1c7409d1-3375-47fb-a93a-0c4da4974753) - Applied Intuition - Sunnyvale, CA - $126k – $423k/yr
- [ML Perception Software Engineer](https://hotfix.jobs/jobs/13c384e6-7a36-4b3b-ab53-5a84170b4d04) - Applied Intuition - Sunnyvale, CA - $125k – $222k/yr
- [Software Engineer - Prediction and Behavior ML](https://hotfix.jobs/jobs/d8d6dc59-bef9-4aea-87d3-9bde2cf94bd1) - Applied Intuition - Sunnyvale, CA - $125k – $222k/yr

**Apply:** https://hotfix.jobs/jobs/835ab781-11d3-499a-8ea6-edaf39d23e77
**Canonical:** https://hotfix.jobs/jobs/835ab781-11d3-499a-8ea6-edaf39d23e77