Sign up to access all features of our service.
  • Job search
  • Favorites
  • Create a CV
    New
  • Salaries
  • Subscriptions

Staff Software Engineer - ML Infra

$232k

Uber

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

Vacancy posted 4 days ago
Similar jobs that could be interesting for youBased on the Staff Software Engineer - ML Infra in Sunnyvale, CA vacancy
  • $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... 
    Suggested
    Full time

    Nvidia

    Santa Clara, CA
    8 hours ago
  • $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... 
    Suggested
    For contractors
    Work at office
    Flexible hours

    Linkedin

    Sunnyvale, CA
    3 days ago
  • $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... 
    Suggested
    Immediate start
    Flexible hours
    Shift work

    Nuro

    Mountain View, CA
    4 days ago
  • $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... 
    Suggested
    Full time
    Contract work
    Temporary work
    Part time
    Local area
    Shift work

    Rivian

    Palo Alto, CA
    4 days ago
  • $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... 
    Suggested
    Full time
    Local area
    Work from home

    News Break

    Mountain View, CA
    4 days ago
  • $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... 

    Google

    Sunnyvale, CA
    4 days ago
  • $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 employment
    Work at office
    Immediate start
    Remote work
    Worldwide
    Relocation
    Flexible hours

    ServiceNow

    Santa Clara, CA
    2 days ago
  • $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 time
    Temporary work
    Part time
    Visa sponsorship

    Walmart

    Sunnyvale, CA
    2 days ago
  • $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 work
    Flexible hours

    Coupang

    Mountain View, CA
    2 days ago
  • $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 time
    Temporary work
    Part time

    Sam's Club

    Sunnyvale, CA
    5 days ago
  • $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

    Plus.ai

    Santa Clara, CA
    5 days ago
  • $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 job
    Work experience placement
    Immediate start
    Flexible hours

    Nuro

    Mountain View, CA
    4 days ago
  • $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... 

    Google

    Mountain View, CA
    3 days ago
  • $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 work
    Work at office
    Immediate start
    Remote work
    Flexible hours

    ServiceNow

    Santa Clara, CA
    9 hours ago
  • $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 time
    Temporary work
    Part time

    Walmart

    Sunnyvale, CA
    2 days ago
  • $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 time
    Temporary work
    Part time

    Walmart

    Sunnyvale, CA
    5 days ago
  • $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... 

    Google

    Sunnyvale, CA
    3 days ago
  • $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... 

    Google

    Mountain View, CA
    2 days ago
  • $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

    Google

    Mountain View, CA
    1 day ago
  • $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 time
    Local area
    Work from home
    Flexible hours

    General Motors

    Sunnyvale, CA
    2 days ago
  • $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... 

    Google

    Mountain View, CA
    2 days ago
  • $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 time
    Temporary work
    Part time
    Remote work

    Walmart

    Sunnyvale, CA
    4 days ago
  • $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... 

    Google

    Mountain View, CA
    4 days ago
  • $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 time
    Temporary work
    Part time
    Immediate start

    Walmart

    Sunnyvale, CA
    3 days ago
  • $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 office
    Local area
    3 days per week

    Aurora Innovation

    Mountain View, CA
    2 days ago
  • $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 start
    Flexible hours

    Nuro

    Mountain View, CA
    4 days ago
  • $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

    Kodiak Robotics

    Mountain View, CA
    7 hours ago
  • $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 placement
    Remote work

    d-Matrix

    Santa Clara, CA
    4 days ago
  • $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 time
    Temporary work
    Part time

    Walmart

    Sunnyvale, CA
    4 days ago
  • $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 time
    Temporary work
    Part time
    Local area

    Walmart

    Sunnyvale, CA
    4 days ago

Do you want to receive more vacancies?

Subscribe and receive similar vacancies to Staff Software Engineer - ML Infra. Be the first to apply!