Research Engineer (AI/RL Infrastructure)

Join Applied Intuition as a Research Engineer (AI/RL Infrastructure) and contribute to cutting-edge technology enabling next-generation physical AI. You will design and build training and evaluation infrastructure, develop large-scale data sampling and curation pipelines, and collaborate closely with AI research, autonomy, and platform teams. Enjoy benefits such as health insurance, a fitness stipend, 401(k) match, a learning stipend, and parental leave.

• 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.
• Collaborate closely with AI research, autonomy, and platform teams to translate cutting-edge research into production-ready systems.
• Experience building and operating production-grade software systems across the full machine learning lifecycle, including training, evaluation, data, 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
• Opinions about building a company-wide platform for ML training, evaluation, and deployment
• Technical experience in: Pytorch, CUDA, Ray, Flyte, K8s
• Deep familiarity with the open-source ML and systems ecosystem, with judgment on when to adopt open source versus build in-house
• Industry experience on relevant topics (self-driving application preferred)
• Don’t meet every single requirement? If you’re excited about this role but your past experience doesn’t align perfectly with every qualification in the job description, we encourage you to apply anyway. You may be just the right candidate for this or other roles