> Markdown version of [/jobs/ext/2710980-staff-software-engineer-ml-infra](https://www.wearedevelopers.com/jobs/ext/2710980-staff-software-engineer-ml-infra). Every page supports `.md` or `Accept: text/markdown`. Links point to the HTML versions so they work for humans too. Agent guide: [/agents.md](https://www.wearedevelopers.com/agents.md).

---

# Staff Software Engineer - ML Infra

- **Company:** Data Inc
- **Location:** Sunnyvale, CA, United States (Remote available)
- **Experience:** Expert
- **Salary:** $232,000.0
- **Contract:** Permanent contract
- **Skills:** Airflow, Batch Processing, Nvidia CUDA, Azure Machine Learning, Graphics Processing Unit (GPU), Kubernetes, Information Technology, TensorRT
- **Published:** September 4, 2026
- **Apply:** https://www.themuse.com/jobs/uber/staff-software-engineer-ml-infra

## About the Role

* Minimum 6 years of working experience.
 * Proven experience for executing technical projects from conception to production.
 * Bachelor's degree in Computer Science or related fields.
 * Familiar with GPU technologies: CUDA, TensorRT, GPU optimizations
 * Familiar with compute cluster management: Kubernetes
 * Familiar with batch orchestration: Flyte or Airflow
 * Familiar with distributed batch processing: Ray

Preferred Qualifications

 * Master or PhD degree in Computer Science or related fields.
 * Background in ML Platform teams from large companies.

## Description

Uber is launching AV Labs to accelerate the autonomous technology ecosystem. We're building out a high-velocity team of multi-disciplinary experts to turn real-world operations into high-quality data for our autonomous partners. This team will be focused on the hardest problem in AV today: unlocking real-world, long-tail driving data. Autonomy is now a data race-and Uber has an edge: We collect rare, real-world driving data at a scale and capital efficiency no one else can match (millions of Uber trips every hour across cities, conditions, and edge cases create the data autonomy has been missing).

We will build platforms that harness scale and real-world complexity to reimagine how the world moves.

You will be a software engineer in AV Labs and involved in the development and implementation of the latest technology for self-driving stacks. The ideal candidate will be able to identify issues, provide solutions, and implement the fixes, as well as set a high technical excellence bar in all things we do.

What the Candidate Will Do:

 * Work with Uber's ML platform and design customized solutions for AV use cases.
 * Design and implement scalable infrastructure on multiple regions and multiple cloud providers.
 * Lead stakeholder alignment with Uber infrastructure teams and AV Labs teams.
 * Drive efficiency efforts on GPUs.


## Related Videos

- [From Messy Queries to Scalable Systems - How Data Engineering actually works](https://www.wearedevelopers.com/videos/100203-from-messy-queries-to-scalable-systems-how-data-engineering-actually-works)
- [Tour de Force: Open-Source LLM Inference Optimization from Simple to Sophisticated](https://www.wearedevelopers.com/videos/100099-tour-de-force-open-source-llm-inference-optimization-from-simple-to-sophisticated)
- [Shipping Faster with Less: Render on Cloud Hosting, AI Workloads, and the Future of DevOps](https://www.wearedevelopers.com/videos/1894-shipping-faster-with-less-render-on-cloud-hosting-ai-workloads-and-the-future-of-devops)
- [Understanding Kubernetes in a visual way](https://www.wearedevelopers.com/videos/100085-understanding-kubernetes-in-a-visual-way)
- [Efficient deployment and inference of GPU-accelerated LLMs​](https://www.wearedevelopers.com/videos/929-efficient-deployment-and-inference-of-gpu-accelerated-llms)
- [Introduction to Azure Machine Learning](https://www.wearedevelopers.com/videos/368-introduction-to-azure-machine-learning)

## Related Articles

- [Highest Paying Tech Companies for Developers](https://www.wearedevelopers.com/magazine/220-highest-paying-tech-companies-for-developers)
- [How to Become an AI Engineer](https://www.wearedevelopers.com/magazine/331-how-to-become-an-ai-engineer)
- [Fully Remote Software Engineer Jobs](https://www.wearedevelopers.com/magazine/447-fully-remote-software-engineer-jobs)
- [The Most Popular IT Jobs on the Market](https://www.wearedevelopers.com/magazine/376-the-most-popular-it-jobs-on-the-market)
- [Top-Paying Tech Jobs (with Salaries)](https://www.wearedevelopers.com/magazine/372-top-paying-tech-jobs-with-salaries)
- [Data Engineer Salary UK](https://www.wearedevelopers.com/magazine/253-data-engineer-salary-uk)