- Robot type
- Autonomous Vehicle
- Location
- AustinTexasUSA
- Job type
- Artificial Intelligence
- Posted
- Jul 29, 2026
Lead AI Infrastructure Engineer
Job description
Our team is at the core of Avride's self-driving stack. We build the base infrastructure layer that powers all autopilot code. It includes a C++ framework for implementing autonomy components, execution graph building and optimization systems, as well as runtimes that execute those graphs, both onboard and in simulation.
The vast part of the execution graph is implemented as a chain of neural network operations. The onboard mode relies on stable latencies of the inference of those networks, while in simulation we also optimize throughput at scale.
Job responsibilities
- At first, you will take on the GPU inference framework, focusing on performance
- Later, the role assumes responsibility and ownership for broader ML infrastructure scattered across ML pipelines
- Close collaboration with the applied ML team responsible for defining the neural model's architecture
Job requirements
- Experience with PyTorch
- Understanding of how GPUs work
- Experience in diagnosing and resolving performance issues
- Strong record of building infrastructure including distributed systems
- 5+ years of experience with C++
- Programming experience in multi-threaded environments - multiple processes, threads, timers, and interrupts
Similar jobs
Avride · Artificial Intelligence
Software Engineer – ML Platform
Austin, Texas
Avride · Artificial Intelligence
Lead Data Scientist – Autonomous Driving
Austin, Texas
Avride · Artificial Intelligence
Machine Learning Engineer – Motion Planning & Prediction
Austin, Texas
Avride · Artificial Intelligence
Machine Learning Engineer
Austin, Texas
