Staff Software Engineer - ML Infra
$232kUber
About The Role 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. Basic Qualifications 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. For Sunnyvale, CA-based roles: The base salary range for this role is USD $232,000 per year - USD $258,000 per year. You will be eligible to participate in Uber's bonus program, and may be offered an equity award & other types of comp. All full‑time employees are eligible to participate in a 401(k) plan. You will also be eligible for various benefits. Ready to Ride? This isn't the kind of place where you follow a playbook - it's where you help write one. If you're driven by impact, energized by challenge, and ready to shape how the world moves - we'd love to hear from you. You may be eligible for bonuses, equity, and other compensation, as well as a range of benefits. Explore our benefits. Offices remain key to collaboration and Uber's culture. Unless approved for full remote work, employees must spend at least 50% of their time in‑office. Some roles, like those at greenlight hubs, require full‑time in‑office presence. Ask your Recruiter for details about this role's requirements. Uber is proud to be an Equal Opportunity employer. All qualified applicants will receive consideration for employment without regard to sex, gender identity, sexual orientation, race, color, religion, national origin, disability, protected Veteran status, age, or any other characteristic protected by law. We also consider qualified applicants regardless of criminal histories, consistent with legal requirements. If you have a disability or special need that requires accommodation, please let us know by completing this form. #J-18808-Ljbffr Uber
$272k - $431.25k
We are seeking a Principal AI and ML Infra Software Engineer, GPU Clusters at NVIDIA to join our Hardware Infrastructure team. As an Engineer, you will have a pivotal role in enhancing efficiency for our researchers by implementing progressions throughout the entire stack...SuggestedFull time$198k - $326k
...model training, feature engineering and serving with... ...scale feature engineering infra for all AI use cases from... ...frameworks, data infra, compute software, and hardware to... ...with hundreds of new ML models per quarter using... ...inference at scale.As a Sr. Staff Software Engineer, you...SuggestedFor contractorsWork at officeFlexible hours$193.93k - $352.29k
...investors.About the RoleWe are looking for a Senior/Staff Software Engineer to serve as a technical leader for Nuro’s ML Data engine. You will sit at the critical... ...Leadership: Work across autonomy teams and data infra teams to build effective ML data pipelines and products...SuggestedImmediate startFlexible hoursShift work$206.5k - $258.1k
...future generations. Role Summary Rivian Autonomy is building an ML Infrastructure team to give hundreds of ML engineers a training platform they can trust at fleet scale. We are seeking a Staff Software Engineer to help design and build the platform that trains and evaluates...SuggestedFull timeContract workTemporary workPart timeLocal areaShift work$180k - $225k
...visit About the role:We’re hiring a Machine Learning Infrastructure Engineer to help build the backbone that trains, serves, and monitors the... ...from an idea to a safely launched model.You’ll work across the ML lifecycle: making training faster and more reliable, improving...SuggestedFull timeLocal areaWork from home$147k - $210k
...computing concepts, consensus algorithms (e.g., Paxos).Google's software engineers develop the next-generation technologies that change how... ...performant bulk data loading pathways(essential for emerging AI/ML workloads), introducing advanced mutation semantics and data life...$190.9k - $334.1k
Company DescriptionIt all started when engineer Fred Luddy wrote code that automated a... ...intelligence.We are looking for a Senior Staff IAM Software Engineer to bring full stack and... ...Experience with graph databases (Neo4j) or ML-assisted role mining and access recommendation...Permanent employmentWork at officeImmediate startRemote workWorldwideRelocationFlexible hours$143k - $286k
...data, and platform teamsMentor engineers and lead design and code... ...Qualifications 8+ years of professional software engineering experienceStrong... ...with AI/ML integration and data-driven systemsExperience... ...StockSummaryWe are seeking a Staff Software Engineer to design and...Full timeTemporary workPart timeVisa sponsorship$152k
...world. Role Overview The Resource Fabric Engineering team builds and operates the... ...innovation at scale. We are looking for a Staff Software Engineer to lead the design and evolution... ...and Compliance teams to define end-to-end ML model development lifecycle experiences....Temporary workFlexible hours$143k - $286k
...Sam's ClubBusiness Segment: Home OfficeRole summary:The (USA) Staff, Software Engineer plays a critical role in delivering scalable platform... ...applies cloud-native architecture principles, and integrates AI/ML solutions to enhance platform performance. The role involves...Full timeTemporary workPart time$130k - $220k
Santa Clara, CASoftware Engineering - Motion Planning /Full-time /HybridPlusAI is a Physical AI... ...pioneering AI-based virtual driver software for factory-built autonomous trucks. Headquartered... ...the production runtime pipeline for ML-based autonomous vehicle planning.Integrate...Full time$176.4k - $319.72k
...Price, and other leading investors.About the RoleAs a Senior/Staff Software Engineer working on driving behavior verification, you are... ...autonomous vehicle.Leverage a combination of machine learning (ML) models and safety metrics from literature to evaluate the end...Odd jobWork experience placementImmediate startFlexible hours$262k - $364k
...practical experience. 8 years of experience in software development.7 years of experience leading technical project strategy, ML design, and optimizing industry-scale ML... ...Preferred qualifications:Master’s degree or PhD in Engineering, Computer Science, or a related technical...$190.9k - $334.1k
...DescriptionIt all started when engineer Fred Luddy wrote code that... ...DescriptionWe are seeking a Senior Staff Engineer (IC5) who will set... ...defining approaches for novel AI/ML challenges, establishing best... ...role you have:14+ years of software development experience with a...Temporary workWork at officeImmediate startRemote workFlexible hours$110k - $220k
...WalmartBusiness Segment: Home OfficeRole summary: Walmart is seeking a Staff Software Engineer to lead the development and delivery of scalable platform capabilities, including infrastructure, tooling, and AI/ML frameworks. This role involves driving the full software...Full timeTemporary workPart time$143k - $286k
...PermanentCompany: WalmartBusiness Segment: Home OfficeRole summary:As a Staff Software Engineer at Walmart, you will lead the design and delivery of... ...lifecycle with a focus on cloud-native architecture and AI/ML integration. You will provide technical leadership, mentor...Full timeTemporary workPart time$207k - $300k
...and efficiency metrics to identify bottlenecks and engineer scalable optimizations across Google's infrastructure... ...equivalent practical experience.8 years of experience in software development.5 years of experience with ML design and ML infrastructure (e.g., model deployment...$207k - $300k
...degree or equivalent practical experience.8 years of experience in software development.5 years of experience testing and launching... ...optimization, personalization, or recommender systems.Google's software engineers develop the next-generation technologies that change how...$262k - $364k
...cross-system bottlenecks; raise the bar on engineering excellence, experimentation velocity, and... ...experience.8 years of experience with software development in one or more programming languages... ...architecture.8 years of experience with ML/AI algorithms and tools, deep learning or...Worldwide$189.3k - $290.7k
Job DescriptionJob Summary We are seeking a Staff Software Engineer to join our ADAS HMI team within the Interactive Experience platform. In this... ...models on edge or embedded devices. Experience with ML frameworks and edge-inference runtimes such as PyTorch, TensorFlow...Full timeLocal areaWork from homeFlexible hours$262k - $364k
Leverage AI/ML advancements, including LLMs and Agentic AI, to solve critical optimisation... ...world of AI.Identify gaps in existing engineering processes and solve using AI-driven... ...equivalent practical experience.8 years of software development experience in one or more programming...$143k - $286k
...options are not available. About the Role:Walmart is seeking a Staff Software Engineer to lead the development and delivery of scalable platform capabilities, including infrastructure, tooling, and AI/ML frameworks. This role involves driving the full software lifecycle...Full timeTemporary workPart timeRemote work$207k - $300k
...Bachelor's degree in Computer Science, Computer Engineering, Electrical Engineering, Applied... ...practical experience.8 years of experience in software development.Experience in Python and C++,... ...profiling workloads using standard ML profilers (e.g., PyTorch profiler) and internal...$143k - $286k
...PermanentCompany: WalmartBusiness Segment: Home OfficeRole summary:As a Staff Software Engineer at Walmart, you will lead the design and delivery of... ...lifecycle with a focus on cloud-native architecture and AI/ML integration. You will provide technical leadership, mentor...Full timeTemporary workPart timeImmediate start$189k - $303k
...and accessible for all. We’re searching for a Staff Machine Learning Engineer - Online Maps to join Aurora’s Maps ML team. The Maps ML team is responsible for providing... ...Aurora DriverRequired QualificationsExcellent software engineering skills in Python and/or...Work at officeLocal area3 days per week$213.48k - $319.72k
...destinations safely and efficiently. We continuously refine our routing engine to calculate more efficient routes, deliver highly accurate ETAs,... ...in C++ programmingExperience in training and inferencing ML models.Experience with map data and graph data structuresExperience...Immediate startFlexible hours$240k - $265k
...leveraging its commercial self-driving software to develop, test and deploy autonomous... ...Department of Defense.We are looking for a Staff Software Engineer to help shape how learned models are... ...Machine Learning, working closely with ML engineers and autonomy teams to bring learned...Visa sponsorship$190k - $300k
...technology. We are at the forefront of software and hardware innovation, pushing the boundaries... .... Will consider remote.The Role: Staff Software Engineer, SIMD KernelsWhat you will do:As a... ...enhance, and maintain software kernels for ML operators—such as softmax, layer norm,...Work experience placementRemote work$143k - $286k
...WalmartBusiness Segment: Home OfficeRole summary: The (USA) Staff, Software Engineer plays a critical role in delivering scalable platform capabilities... ...requires expertise in cloud-native architecture, AI/ML integration, and software security to build robust, modular...Full timeTemporary workPart time$143k - $286k
...Segment: Home OfficePlatform Engineering Execution: Delivers platform... ...infrastructure, tooling, and ML frameworks (for example, CI/CD... ...and agility across systems. Software Development & Lifecycle Ownership... ...include : - StockSummaryAs a Staff Software Engineer on the...Full timeTemporary workPart timeLocal area
Do you want to receive more vacancies?
Subscribe and receive similar vacancies to Staff Software Engineer - ML Infra. Be the first to apply!
- government software Sunnyvale, CA
- software applications developer Sunnyvale, CA
- healthcare software sales Sunnyvale, CA
- software technical support engineer Sunnyvale, CA
- software support Sunnyvale, CA
- embedded software Sunnyvale, CA
- remote software sales Sunnyvale, CA
- software intern Sunnyvale, CA
- software technical support Sunnyvale, CA
- software engineer - cloud services Sunnyvale, CA

