# Lead AI Infrastructure Engineer

**Company:** [Avride](https://hotfix.jobs/companies/avride)
**Location:** Austin, TX
**Role:** ML Engineering
**Experience:** 5+ years
**Skills:** PyTorch, C++, GPU, Distributed Systems, multi-threading
**Posted:** 2026-07-30

> Lead the ML infrastructure layer with focus on optimizing GPU inference performance for real-time onboard and high-throughput offboard autonomous driving applications. Requires deep experience with PyTorch, C++, GPUs, distributed systems, and performance optimization.

## Job Description

## What you'll do
- Take ownership of the GPU inference framework with a focus on performance for both onboard (near real-time) and offboard (high throughput, deterministic) applications.
- Assume responsibility for broader ML infrastructure across ML pipelines.
- Collaborate closely with the applied ML team on neural model architectures.

## What you'll need
- Experience with PyTorch.
- Strong understanding of GPU architecture and operation.
- Experience diagnosing and resolving performance issues.
- Strong record building infrastructure, including distributed systems.
- 5+ years of experience with C++.
- Programming experience in multi-threaded environments (multiple processes, threads, timers, interrupts).

## Similar roles

- [Senior Machine Learning Engineer, Price Modeling](https://hotfix.jobs/jobs/ebda518f-cc52-48e2-a526-6c814f4334b3) - Airbnb - Remote - $196k – $227k/yr
- [Senior Software Engineer, Autonomy Capabilities - Space](https://hotfix.jobs/jobs/9cca1cd3-4d50-4638-95c5-233aa6fc65a5) - Shield AI - Boston, MA - $160k – $240k/yr
- [Senior Manager, Machine Learning Engineering-Applied Research](https://hotfix.jobs/jobs/c1fa1d53-0f95-41ef-a92d-f0aca67444ca) - Pinterest - Remote - $228k – $469k/yr
- [Senior AI Engineer, Agentic Data Enrichment](https://hotfix.jobs/jobs/cd9cf504-6eed-41f6-95fe-6dd1b17fd649) - Baselayer - San Francisco, CA - $230k – $340k/yr
- [Lead Software Platform Engineer, MLOps](https://hotfix.jobs/jobs/16f5892f-2ec3-431e-afd3-9f137718589b) - TetraScience - Remote

**Apply:** https://hotfix.jobs/jobs/f759cd0f-5533-451b-9e56-6875a22b5ae3
**Canonical:** https://hotfix.jobs/jobs/f759cd0f-5533-451b-9e56-6875a22b5ae3