Research Engineer - AI/RL Infrastructure

Job summary

Responsibilities

We are looking for a passionate Research Engineer (AI/RL Infrastructure) to join the Research Group at Applied Intuition. This role is ideal for engineers who design, build, and operate state-of-the-art, large-scale ML systems and enjoy working closely with researchers to develop and accelerate…

• Design and build training and evaluation infrastructure to support our current 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

What they're looking for

• 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

Tech stack

• PyTorch
• CUDA
• Ray
• Kubernetes