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

Senior/Staff Software Engineer, ML Infrastructure, Optimization

$193.93k - $352.29k
Full-time

Nuro

Who We Are

Nuro believes self-driving vehicles are the most immediate and profound opportunity for AI to drive positive change in the physical world. Safer streets, more time for what matters, and easier access to the world around us, that’s why we’re building a universal autonomy platform: self-driving for all roads and all rides.

Founded in 2016, Nuro is a physical AI company developing Level 4 autonomous driving technology for a wide range of vehicles, use cases, and markets. Powered by the Nuro Driver™, our universal autonomy platform enables the global mobility ecosystem to deploy autonomy at scale, from robotaxis and logistics fleets to personal vehicles.

With years of real-world deployment experience and a flexible, partner-led business model, Nuro is working toward a future where millions of autonomous vehicles powered by our technology help make everyday life safer, easier, and more connected.

Nuro has raised over $2B in capital from Uber, NVIDIA, Google, Softbank, Fidelity, T. Rowe Price, and other leading investors.

About The Role

The Autonomy ML Infrastructure team is responsible for building & improving the core infrastructure for autonomy teams at Nuro. In this role, you will work closely with teams across Nuro, to design, build and deploy core infrastructure components in machine learning model life cycle, to push the autonomous future forward. You will have an opportunity to work across the full stack of machine learning solutions - from designing robust and scalable model pipelines to building to deploying the optimized models on Nuro’s fleet of self-driving robots!

About The Work

  • Optimize Nuro’s autonomy stack with cutting-edge optimization techniques like quantization, distillation, and model compression.
  • Work with autonomy engineers to optimize, validate, and deploy large language models.
  • Develop and maintain a world-class model compiler framework, FTL.
  • Write robust, high quality software to increase our confidence in our vehicle’s ability to navigate safely on-road.
  • Collaborate closely with machine learning domain experts and engineers across behavior, perception and mapping to design and implement end-to-end learned ML solutions.

About You

  • 3+ years of relevant experience in ML optimization infrastructure.
  • Experience with ML optimization techniques such as quantization and pruning, and ML compilers.
  • Experience maintaining, profiling, and optimizing GPU ML compilers & runtimes.
  • Proficient in Python and working experience with C++ and CUDA.
  • Working experience deep learning frameworks (like PyTorch, Jax, Tensorflow, Keras).
  • Proficient in Python and working experience with C++.
  • You are passionate about accelerating the benefits of robotics for everyday life.

At Nuro, your base pay is one part of your total compensation package. For this position, the reasonably expected base pay range is between $193,930 and $352,290 for the level at which this job has been scoped. Your base pay will depend on several factors, including your experience, qualifications, education, location, and skills. In the event that you are considered for a different level, a higher or lower pay range would apply. This position is also eligible for an annual performance bonus, equity, and a competitive benefits package.

At Nuro, we celebrate differences and are committed to a diverse workplace that fosters inclusion and psychological safety for all employees. Nuro is proud to be an equal opportunity employer and expressly prohibits any form of workplace discrimination based on race, color, religion, gender, sexual orientation, gender identity or expression, national origin, age, genetic information, disability, veteran status, or any other legally protected characteristics.
Vacancy posted 25 days ago
Similar jobs that could be interesting for youBased on the Senior/Staff Software Engineer, ML Infrastructure, Optimization in California vacancy
  • $254k - $350k

     ...Zoox ML Performance Optimization Team Opportunity Zoox is on a mission to reimagine transportation...  ...work alongside a team of strong software engineers and act as a force multiplier for...  ...want to learn more about our ML Infrastructure, here is one of our past talks at... 
    Senior

    Zoox

    San Mateo, CA
    1 day ago
  • $193.93k - $352.29k

     ...fungible is the infrastructure that decides whether...  ...Nuro's own engineering organization, under...  ...You ~5+ years of software engineering experience...  ...experience. Staff-level candidates...  ...tuning, preference optimization, and RL — and you...  ...Experience with ML training or research... 
    Senior
    Full time
    Immediate start
    Flexible hours

    Nuro

    California
    12 days ago
  • $210k - $300k

     ...obsessed team of the world's best engineers and operators. If you are...  ...the Role We're looking for a Software Engineer to join our ML Infrastructure team. In this role, you'll help build...  ...workloads. Develop, tune, and optimize low-level CUDA kernels. Design... 
    Senior
    Local area
    Flexible hours

    Nimble Robotics

    San Francisco, CA
    27 days ago
  • $208.85k - $271.52k

     ...Job Summary   As a Senior Staff Software Engineer focused on AI platforms, you...  ..., data, and computational infrastructure. You will architect scalable...  ...automated testing, performance optimization, and maintainable code...  ...Experience integrating AI/ML services, LLM APIs, intelligent... 
    Senior
    Work at office
    Local area

    Murata America

    San Diego, CA
    more than 2 months ago
  • $254k - $350k

     ...Boston, MA / Seattle, WASoftware - Autonomy Software /Full-time /HybridThe Perception team...  ....As a Machine Learning and System Optimization Engineer, you will orchestrate and allocate overall...  ...SoCs.In addition, you will optimize ML models, write custom CUDA kernels, and... 
    Senior
    Full time
    Temporary work
    Relocation package

    Zoox

    San Mateo, CA
    4 days ago
  • $193.93k - $352.29k

     ...investors. About the Role Our software team is growing, and we are looking for talented engineers to join us and be...  ...Platform, Simulation, and Technical Infrastructure. Data Platform: The Data...  ...performance tuning and optimization, robotics software frameworks... 
    Senior
    Immediate start
    Flexible hours

    Nuro

    Mountain View, CA
    25 days ago
  •  ...operations, systems and safety engineering – all dedicated to redefining...  ...road, ensuring that they are optimized for onboard performance and...  ...dedicated to developing cutting-edge ML algorithms and models that...  ...maintain industry-leading software practices and principles -... 
    Senior
    Full time

    Latitude

    Palo Alto, CA
    a month ago
  •  ...About the Role You'll build the software backbone that powers autonomous manufacturing...  ...operations at Tenkara. As a Senior/Staff Software Engineer on the Platform team, you'll design...  ...productionize LLM-based agents, AI/ML infrastructure, and RAG patterns for intelligent... 
    Senior
    Full time

    Tenkara

    San Francisco, CA
    a month ago
  • $190.9k - $334.1k

     ...all started when engineer Fred Luddy wrote...  ...We are seeking a Senior Staff Engineer  (IC5) who...  ...for novel AI/ML challenges, establishing...  ...platforms and infrastructure that enable AI-...  ...organization Software Development Excellence...  ..., retrieval optimization, and vector database... 
    Senior
    Full time
    Temporary work
    Work experience placement
    Work at office
    Immediate start
    Remote work
    Flexible hours

    ServiceNow

    Santa Clara, CA
    22 days ago
  • $193.93k - $352.29k

     ...autonomous driving technology. In an ML-first system, the overall...  ...a scalable and reliable data infrastructure. This infrastructure is...  ...collaborates closely with system engineers to thoroughly validate the...  ...best practices across broader software organization A bachelor's... 
    Senior
    Full time
    Work experience placement
    Immediate start
    Flexible hours

    Nuro

    California
    11 days ago
  • $160k - $200k

     ...Premier Global Links LLC is hiring a Senior Software Engineer Optimization to develop software that combines...  ...mathematical optimization and AI/ML. This position is primarily backend...  ...workflows and scalable application infrastructure. Convert mathematical and optimization... 
    Senior
    Full time
    Remote work

    Premier Global Links

    San Francisco, CA
    1 day ago
  • $198k - $326k

     ...centered on trust and optimized for culture,...  ...frameworks that empower engineers and researchers...  ...highly scalable infrastructure that delivers...  ...at scale.As a Sr. Staff Software Engineer, you...  ...distributed systems and ML platform problems...  ...level: Mid-Senior LevelIndustry: Internet
    Senior
    For contractors
    Work at office
    Flexible hours
    Shift work

    Linkedin

    Sunnyvale, CA
    1 day ago
  • $193.93k - $352.29k

     ...autonomous driving technology. In an ML-first system, the overall...  ...a scalable and reliable data infrastructure. This infrastructure is...  ...collaborates closely with system engineers to thoroughly validate the...  ...best practices across broader software organization A bachelor's... 
    Senior
    Full time
    Work experience placement
    Immediate start
    Flexible hours

    Nuro

    California
    11 days ago
  •  ...About the Role As a Staff Infrastructure Engineer on the Compute & Storage Platform...  ...compute pipelines. Cost Optimization & Performance Engineering:...  ...Terraform, CloudFormation, etc.). Software Development: Fluency in...  ...Datadog, Prometheus AI/ML Infrastructure support... 
    Work at office
    2 days per week

    Toast Inc

    Dublin, CA
    5 days ago
  • $198k - $326k

     ...centered on trust and optimized for culture,...  ...our world-class software engineering team, you will take...  ...next-generation infrastructure and platforms for...  ...best-in-class AI/ML infrastructure, Kubernetes...  ...company.As a Sr. Staff Software Engineer...  ...level: Mid-Senior LevelIndustry:... 
    Senior
    For contractors
    Work at office
    Flexible hours

    Linkedin

    Sunnyvale, CA
    1 day ago
  • $262k - $364k

     ...modeling and supporting infrastructure that yield our content...  ...of experience with software development in one or...  ...years of experience with ML/AI algorithms and...  ...strategy, ML design, and optimizing ML infrastructure (e.g...  ...tools.Google's software engineers develop the next-... 
    Senior

    Google

    Mountain View, CA
    2 days ago
  • $262k - $364k

     ...coach a distributed engineering team, fostering...  ...deploy scalable software and advanced serving...  ...architectures, optimizing the LLM serving...  ...accelerate HPC, RDMA, and ML workloads....  ...Machine Learning Infrastructure.Google's software...  ...(GDS).As a Senior Staff Software Engineer... 
    Senior
    Remote work
    Worldwide

    Google

    Sunnyvale, CA
    3 days ago
  • $262k - $364k

     ...Deliver reproducible evaluation infrastructure and automated test...  ...patterns.Build the metrics engines and dashboards to track evaluation...  ...8 years of experience in software development.7 years of...  ...project strategy, ML design, and optimizing industry-scale ML infrastructure... 
    Senior

    Google

    Sunnyvale, CA
    2 days ago
  • $153.7k - $248.6k

     ...an AI-Native Cloud FinOps Engineer , you will build...  ...pipelines and autonomous optimization systems that control our cloud...  ...will focus heavily on AI/ML compute infrastructure like GPUs and LLM pipelines...  ...efficiency as a continuous software feature, you will leverage... 
    Senior
    Full time
    H1b
    Work at office
    Local area
    Shift work

    Palo Alto Networks

    Santa Clara, CA
    4 days ago
  • $157.5k - $225k

     ...Zscaler.RoleWe are looking for a Senior Staff Software Engineer - Test Automation Infrastructure (ZIA Core) to join our team....  ...driving continuous regression optimization.What you’ll do (Role Expectations...  ...Foundational understanding of AI/ML technologies and experience... 
    Senior
    Full time
    Work at office
    Local area
    3 days per week

    Zscaler

    San Jose, CA
    2 days ago
  • $214.1k - $309.8k

     ...We partner closely with Engineering, Product, Security, and Quality...  .... Your Impact As a Senior Staff Software Engineer, you will define and...  ...and high-performance test infrastructure. ~1+ years experience using...  .... Experience with AI/ML testing, LLM APIs, prompt engineering... 
    Senior
    Full time
    Contract work
    Temporary work
    Local area
    Flexible hours

    Cisco

    San Jose, CA
    1 day ago
  • $264k - $310k

     ...so are the rewards. The Developer Infrastructure org is the engine behind Robinhood's entire...  ...output of every engineer. As the Senior Staff Software Engineer, Developer Infrastructure...  ...cluster, drive language-agnostic build optimization across Python, Go, TypeScript, and... 
    Senior
    Work at office
    Remote work
    Flexible hours
    Shift work
    3 days per week

    Robinhood

    Menlo Park, CA
    28 days ago
  •  ...Snowflake AI Search Infrastructure Engineer At Snowflake, we are powering the era of the agentic...  ...power Snowflake Cortex. This involves optimizing how billions of rows of data are indexed...  ..., or backend infrastructure for AI/ML products. Technical Stack: Deep proficiency... 

    Snowflake

    Menlo Park, CA
    4 days ago
  • $251k - $310k

     ...development, system design, and software implementation to solve...  ...of computer science, optimization, and fleet management. Instrument...  ..., Data Science, and other Engineering teams to develop and...  ...coding experience. Backend or infrastructure preferred. ~ Experience with... 
    Full time
    Remote work

    Waymo

    California
    13 days ago
  • $229k - $343k

     ...services. We're looking for a Staff Software Engineer to join the Platform...  ...powered testing tools and infrastructure across Snapchat's mobile...  ...) and performance/latency optimization. Deep experience with...  ...effectively with Product, DS, ML, Design, and other... 
    Live in
    Work at office
    Local area

    Snapchat

    Los Angeles, CA
    2 days ago
  • $219k - $233k

     ...Insitro seeks a Staff Software Engineer to architect and scale computer vision and machine learning infrastructure for high-throughput imaging workflows in drug discovery. The role involves...  ..., and partnering with scientists and ML teams to integrate imaging and automation... 
    Senior

    Insitro

    South San Francisco, CA
    4 days ago
  • $193.93k - $291.15k

     ...Senior Software Engineer, ML Infrastructure Mountain View, California (HQ) Nuro believes self-driving vehicles are the most immediate and profound...  .... Intelligent Scheduling: Designing and optimizing workload orchestration to maximize hardware utilization... 
    Senior
    Immediate start
    Flexible hours

    Nuro

    Mountain View, CA
    5 days ago
  • $120k - $200k

     ...more. About the Role As a Senior Infrastructure Engineer at Bland, you'll help us to build the...  ...real-time voice processing, scale ML inference, and integrate with enterprise...  ...ML infrastructure : Create and optimize the infrastructure supporting our AI... 
    Senior
    Work at office
    Night shift

    Bland Company

    San Francisco, CA
    17 hours ago
  • $216k - $337k

     ...implement, test, and deploy motion planning and trajectory optimization algorithms for humanoid robots Architect motion planning systems...  ...robotics systems, owning architecture, and mentoring senior engineers Bonus Qualifications Experience working with humanoid (or... 
    Senior
    Full time
    Temporary work
    Remote work
    Relocation package
    Flexible hours

    Agility Robotics

    Sacramento, CA
    3 days ago
  •  ...Moveworks’ Reasoning Engine and natural language...  ...The Role Are you a software engineer who has honed...  ...for you. As a senior staff software engineer on...  ...code execution, latency optimization, agent memory, LLM self...  ...fast on scalable infrastructure, solving challenging... 
    Senior
    Full time
    Work at office
    Immediate start
    Remote work
    Flexible hours

    ServiceNow

    Mountain View, CA
    4 days ago

Do you want to receive more vacancies?

Subscribe and receive similar vacancies to Senior/Staff Software Engineer, ML Infrastructure, Optimization. Be the first to apply!