AI/RL Infrastructure Engineer for Large-Scale ML Systems

Applied Intuition; Sunnyvale (CA); Hybrid; USD 126,000 - 423,000; Full time

Job summary

Applied Intuition is seeking a Research Engineer (AI/RL Infrastructure) to join the Research Group. You will design, build, and operate state-of-the-art ML systems and collaborate with researchers to accelerate the core platform powering next-generation physical AI, including autonomous driving and robotic generalist.

You will access massive GPU clusters, contribute to production-ready systems, and apply cutting-edge research to real-world deployments across autonomous driving, robotics, and

Qualifications

• Experience building and operating production-grade software systems across the full ML lifecycle.
• 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.
• Strong systems-level debugging skills for distributed training environments.
• Deep familiarity with the open-source ML and systems ecosystem; judgement on open source vs in-house.
• Technical experience in PyTorch, CUDA, Ray, Flyte, and Kubernetes.

Responsibilities

• Design and build training and evaluation infrastructure to support our AI research directions, orchestrating massive GPU clusters to process PBs of multimodal sensor data.
• Build robust benchmarking, continuous evaluation, and regression tracking systems for model performance across diverse driving distributions.
• Develop large-scale data sampling, dataset generation, and data curation pipelines powering a closed-loop data flywheel.
• Enable high-throughput distributed training across heterogeneous cloud environments with a focus on reliability and cost-efficiency.
• Collaborate with AI research, autonomy, and platform teams to translate research into production-ready systems.

Skills

ML systems

Distributed training

Performance tuning

Tools

Pytorch

CUDA

Ray

Flyte

K8s

Job description

Applied Intuition is seeking a Research Engineer (AI/RL Infrastructure) to join the Research Group. You will design, build, and operate state-of-the-art ML systems and collaborate with researchers to accelerate the core platform powering next-generation physical AI, including autonomous driving and robotic generalist.

You will access massive GPU clusters, contribute to production-ready systems, and apply cutting-edge research to real-world deployments across autonomous driving, robotics, and