Avride Logo

Avride

Lead AI Infrastructure Engineer

Posted Yesterday
Be an Early Applicant
In-Office
Austin, TX, USA
Senior level
In-Office
Austin, TX, USA
Senior level
Lead ML infrastructure engineer owning GPU inference performance for onboard real-time and high-throughput offboard systems. Improve and maintain C++ inference framework, optimize GPU execution, collaborate with applied ML teams, and expand ownership across ML pipelines and distributed infrastructure.
The summary above was generated by AI
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].

HQ

Avride Austin, Texas, USA Office

8605 Cross Park Dr, Austin, TX , United States, 78754

Similar Jobs

9 Days Ago
Remote or Hybrid
United States
150K-170K Annually
Senior level
150K-170K Annually
Senior level
Information Technology • Database • Consulting
Lead design and delivery of agentic AI systems and LLMOps on AWS to generate, validate, and deploy infrastructure-as-code (Terraform). Build multi-agent applications, RAG pipelines, and AIOps for cloud operations; integrate AI into CI/CD and ITSM. Set technical direction, establish guardrails/policy-as-code, curate reusable Terraform modules, and mentor engineering teams.
Top Skills: AiopsAmazon Bedrock AgentsArtifactoryAWSCi/CdGitItsmJenkinsLangchainLlmopsRagSonarqubeTerraform
Yesterday
Hybrid
209K-286K Annually
Senior level
209K-286K Annually
Senior level
Fintech • Machine Learning • Payments • Software • Financial Services
Lead design, development, deployment, and support of scalable GenAI platform components (foundation model training, LLM inference, vector search, guardrails, evaluation, observability). Optimize training and inference performance, contribute to technical vision and roadmap, and partner with cross-functional teams to deliver responsible, production-grade AI systems.
Top Skills: Aws UltraclustersC#C++Go (Golang)HuggingfaceJavaLarge Language ModelsLlm InferenceNemo GuardrailsPythonPyTorchScalaSimilarity SearchVectordbs
31 Minutes Ago
Hybrid
Austin, TX, USA
Senior level
Senior level
Fintech • Software • Financial Services
Provide operational and client-facing support for fintech clients by processing daily workflows, identifying and implementing process improvements, analyzing operational data and preparing reports, escalating critical issues, and collaborating across teams to resolve client and processing challenges while enhancing service delivery and client satisfaction.
Top Skills: AIExcelMicrosoft OutlookMicrosoft Word

What you need to know about the Austin Tech Scene

Austin has a diverse and thriving tech ecosystem thanks to home-grown companies like Dell and major campuses for IBM, AMD and Apple. The state’s flagship university, the University of Texas at Austin, is known for its engineering school, and the city is known for its annual South by Southwest tech and media conference. Austin’s tech scene spans many verticals, but it’s particularly known for hardware, including semiconductors, as well as AI, biotechnology and cloud computing. And its food and music scene, low taxes and favorable climate has made the city a destination for tech workers from across the country.

Key Facts About Austin Tech

  • Number of Tech Workers: 180,500; 13.7% of overall workforce (2024 CompTIA survey)
  • Major Tech Employers: Dell, IBM, AMD, Apple, Alphabet
  • Key Industries: Artificial intelligence, hardware, cloud computing, software, healthtech
  • Funding Landscape: $4.5 billion in VC funding in 2024 (Pitchbook)
  • Notable Investors: Live Oak Ventures, Austin Ventures, Hinge Capital, Gigafund, KdT Ventures, Next Coast Ventures, Silverton Partners
  • Research Centers and Universities: University of Texas, Southwestern University, Texas State University, Center for Complex Quantum Systems, Oden Institute for Computational Engineering and Sciences, Texas Advanced Computing Center

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account