Senior/Staff Software Engineer, ML Infrastructure, Optimization
$193.93k - $352.29kFull-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.
- 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.
Vacancy posted 9 days ago
Similar jobs that could be interesting for youBased on the Senior/Staff Software Engineer, ML Infrastructure, Optimization in California vacancy
$193.93k - $352.29k
...About the RoleThe Autonomy ML Infrastructure team is responsible for building... ...building to deploying the optimized models on Nuro’s fleet of... .... Work with autonomy engineers to optimize, validate, and... ...Write robust, high quality software to increase our confidence...SeniorWork experience placementImmediate startFlexible hoursNuro
Mountain View, CA3 days ago$254k - $350k
...Remote (United States)Software - Software Systems /Full... ...this mission. The ML Platform team at Zoox... ...drive our ML Performance Optimization initiatives and make our... ...of strong software engineers and act as a force multiplier... ...more about our ML Infrastructure, here is one of our...SeniorFull timeRemote workZoox
Foster, CA5 days 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...SeniorFull timeTemporary workRelocation packageZoox
San Mateo, CA3 days 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: InternetSeniorFor contractorsWork at officeFlexible hoursShift workLinkedin
Sunnyvale, CA5 days ago$262k - $364k
...technical goal, roadmap, and software architecture for hardware/software co-design, optimizing large-scale ML inference and training... ...optimization across distributed ML infrastructure, balancing compute, memory... ...Computer Science, Computer Engineering, Electrical Engineering, a...SeniorWorldwideGoogle
Sunnyvale, CA2 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:...SeniorFor contractorsWork at officeFlexible hoursLinkedin
Sunnyvale, CA5 days ago$262k - $364k
...coach a distributed team of engineers.Facilitate alignment and... .... 8 years of experience in software development.7 years of experience... ...project strategy, ML design, and optimizing end-to-end Machine Learning... ...is looking for experienced Senior ICs to help bring advanced...SeniorGoogle
Sunnyvale, CA3 days 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-...SeniorGoogle
Mountain View, CA2 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...SeniorRemote workWorldwideGoogle
Sunnyvale, CA2 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...SeniorFull timeWork at officeLocal area3 days per weekZscaler
San Jose, CA1 day ago$254k - $350k
...significant impact. The ML Platform team enables... ...to own ML training optimization at Zoox and make the models... ..., while mentoring engineers and shaping the team’s... ...learn more about our ML Infrastructure, here is one of our... ...tooling, or hardware/software co-design (GPU/TPU/accelerator...SeniorFull time$262k - $364k
Build high-quality serving infrastructure by inventing novel solutions, solving complex problems, and providing technical... ...development and successful launch of high-impact software products, particularly in the AI, ML, or LLM space.Preferred qualifications:Ability to drive...SeniorShift workGoogle
Mountain View, CA2 days ago$193.93k - $352.29k
...investors.About the RoleOur software team is growing, and we are looking for talented engineers to join us and be instrumental... ...Platform, Simulation, and Technical Infrastructure.Data Platform: The Data... ...performance tuning and optimization, robotics software frameworks...SeniorImmediate startFlexible hoursNuro
Mountain View, CA5 days ago$262k - $364k
...experience.8 years of experience in software development.7 years of... ...and developing large-scale infrastructure, distributed systems or... ...systems for large-scale AI/ML workloads.Preferred qualifications... ...:Master’s degree or PhD in Engineering, Computer Science, or a related...SeniorTemporary workGoogle
Mountain View, CA4 days ago$245.4k - $429.45k
...recruiting process here.Sr. Staff Software Engineer, Product ML InfrastructureMillions... ....Pinterest’s Product ML Infrastructure (PMLI) team enables fast... ...standards; mentor senior engineers and raise the... ...execution, kernel and memory optimization, compilation, or quantization...SeniorWork at officeLocal areaRemote workRelocationRelocation packagePinterest
Palo Alto, CA2 days ago$198k - $326k
...on trust and optimized for culture, connection... ..., feature engineering and serving... ..., compute software, and hardware... ...Model Training Infrastructure: As an engineer... ...hundreds of new ML models per quarter... ....As a Sr. Staff Software Engineer... ...level: Mid-Senior LevelIndustry:...SeniorFor contractorsWork at officeFlexible hoursLinkedin
Sunnyvale, CA5 days ago$207k - $300k
Partner with senior leads to architect... ...-time context engineering with model... ...Advance automated optimization methods for... ...experience in software development.5... ...experience with ML design and... ...looking for a Staff Software Engineer... ...and expand our infrastructure, while...Google
Mountain View, CA1 day ago$160k - $280k
...About the Role We're looking for a software engineer to build the Kubernetes-native control... ...packing, right-sizing), and think about GPU infrastructure as software to be engineered. A... ...service. Partner with the inference/ML platform team: understand the cluster shapes...SeniorFull time$262k - $364k
...intersection of core quality, new features, infrastructural optimizations, and model training.Drive team... ...experience.8 years of experience in software development.5 years of experience in... ...and developing the careers of engineering team members.Experience working on user...SeniorGoogle
Mountain View, CA1 day ago$262k - $364k
...Influence and coach a distributed team of engineers.Drive technical project strategy, lead large-scale ML infrastructure optimization, and oversee the design and implementation... ...practical experience.8 years of experience in software development.7 years of experience leading...SeniorGoogle
Mountain View, CA1 day ago$262k - $364k
...deploy, maintain, and enhance large scale software solutions.Provide technical leadership on... ...Influence and coach a distributed team of engineers.Drive technical project strategy, lead large-scale ML infrastructure optimization, and oversee the design and implementation...SeniorGoogle
Sunnyvale, CA3 days ago$262k - $364k
...deploy, maintain, and enhance large scale software solutions.Provide technical leadership on... ...Influence and coach a distributed team of engineers.Drive technical project strategy, lead large-scale ML infrastructure optimization, and oversee the design and implementation...SeniorGoogle
Sunnyvale, CA2 days ago- ...ROLE: AMD is looking for an influential software engineer who is passionate about improving the... ...Able to communicate effectively and work optimally with different teams across AMD. KEY... ...background Knowledge of training and deploying ML models for EDA applications, Data...Senior
AMD
San Jose, CA5 days ago $262k - $364k
...modeling, tuning, and optimization techniques to support... ...years of experience in software development.7 years of... ...technical project strategy, ML design, and optimizing industry-scale ML infrastructure (e.g., model... ...Master’s degree or PhD in Engineering, Computer Science, or...SeniorGoogle
Mountain View, CA4 days ago$262k - $364k
...DeepMind to solve their ML model performance... ...years of experience in software development.7 years of... ...strategy, ML design and optimizing end-to-end Machine Learning... ...etc.) and performance engineering techniques.Experience... ....The AI and Infrastructure team is redefining what...SeniorImmediate startWorldwideGoogle
Sunnyvale, CA2 days ago$262k - $364k
...product area partnerships. Optimize the retail ad products... ...; raise the bar on engineering excellence,... ...years of experience with software development in one or... ...years of experience with ML/AI algorithms and tools... ...strategy, ML design, and ML infrastructure (e.g., model...SeniorWorldwideGoogle
Mountain View, CA2 days ago$262k - $364k
Leverage AI/ML advancements, including LLMs and Agentic AI... ....Identify gaps in existing engineering processes and solve using... ...practical experience.8 years of software development experience in... ...strategy, ML design, and optimizing ML infrastructure (e.g., model deployment, evaluation...SeniorGoogle
Mountain View, CA4 days ago$130k - $220k
Santa Clara, CASoftware Engineering - Motion Planning /... ...-based virtual driver software for factory-built autonomous... ...decision-making and optimization techniques into a... ...today.About the RoleAs a Senior Software Engineer on... ...Partner closely with the ML and prediction teams...SeniorFull timePlus.ai
Santa Clara, CA5 days ago$130k - $220k
Santa Clara, CASoftware Engineering - Motion Planning /Full... ...-based virtual driver software for factory-built autonomous... ..., implement, and optimize the production runtime pipeline for ML-based autonomous vehicle... ...solutions.Build tooling and infrastructure to improve debugging,...SeniorFull timePlus.ai
Santa Clara, CA1 day ago- ...among product areas. As a Senior Staff Software Engineer, you will own broad, ambiguous... ...across AI platform infrastructure. Key areas include harness... ...strong, hands-on engineers. AI/ML background is a strong... ...systems, scaling, latency optimization, reliability engineering,...SeniorWork at office3 days per week
Rippling
San Francisco, CA4 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!
Related searches
- remote infrastructure engineer California
- senior infrastructure engineer California
- infrastructure developer California
- infrastructure engineering manager California
- principal infrastructure engineer California
- infrastructure engineer California
- data infrastructure engineer California
- senior fpga design engineer California
- senior specialist California
- senior advisor California