Mirrai Careers
Resume BuilderCareer Test
InsightsPricing
Get Started Free
Jobs/Lead AI Infrastructure Engineer

Lead AI Infrastructure Engineer

Avride

Austin, Texas Posted 6d ago
Apply on company site
ABOUT THE TEAM 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. ABOUT THE ROLE We’re looking for a software engineer with a leadership mindset and deep ML infrastructure experience. You will decide and influence the ML infrastructure layer across the company. The biggest challenge we’re facing at the moment is the effectiveness of GPU inference - both for onboard applications with near real-time guarantees and for offboard cases that target high throughput and deterministic execution. It is the first priority within this role. WHAT YOU'LL DO * 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 WHAT YOU'LL NEED * 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   Candidates are required to be authorized to work in the U.S. The employer is not offering relocation sponsorship, and remote work options are not available. Avride is an equal opportunity employer and committed to providing reasonable accommodations to qualified applicants and employees with disabilities to ensure they have equal access to employment opportunities. Avride complies with the Americans with Disabilities Act (ADA), if you need a reasonable accommodation to assist with the application or hiring process, or to perform the essential functions of a job, please email [email protected].

See how well you match this job

Upload your resume and we’ll score your fit for this role and 6 similar roles — then tailor your CV to it with AI. Free, no credit card.

Check your match

Similar jobs

  • Lead Software Engineer – Control

    avride

    Austin, Texas
  • Sensor Integration Software Engineer – Autonomous Vehicles

    avride

    Austin, Texas
  • Sr. Machine Learning Ops Engineer

    haydenai

    Remote$200k–$260k
  • Software Engineer, Model Inference

    OpenAI

    San Francisco$295k–$555k
  • Software Engineer, Compute Infrastructure

    OpenAI

    Remote$230k–$405k
  • Staff Software Engineer, Infrastructure

    aerovect

    Atlanta - Onsite
Apply on company site

Want more roles like this? Browse fresh jobs or tailor your resume with AI.

Mirrai Careers

AI-powered career platform: build resumes, match jobs, and plan your career.

Product

  • All Tools
  • Resume Builder
  • Career Test
  • Pricing
  • For employers

Legal

  • Privacy Policy
  • Terms of Service
  • Fair Use Policy

Company

MIRRAI CHAT LTD (Company No. 16403306)

71-75 Shelton Street, Covent Garden

London, WC2H 9JQ, UNITED KINGDOM

[email protected]

© 2026 Mirrai Careers. All rights reserved.