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

Staff Software Engineer, Deep Learning Acceleration

$189k - $274k
Full-time

Aurora

Who We Are

Aurora’s mission is to deliver the benefits of self-driving technology safely, quickly, and broadly.

The Aurora Driver will create a new era in mobility and logistics, one that will bring a safer, more efficient, and more accessible future to everyone.

At Aurora, you will tackle massively complex problems alongside other passionate, intelligent individuals, growing as an expert while expanding your knowledge. For the latest news from Aurora, visit aurora.tech or follow us on LinkedIn.

Aurora hires talented people with diverse backgrounds who are ready to help build a transportation ecosystem that will make our roads safer, get crucial goods where they need to go, and make mobility more efficient and accessible for all. As a Staff Software Engineer focusing on Deep Learning Acceleration at Aurora, you will play a pivotal role in enhancing the performance of Deep Learning networks utilized in our Autonomous Vehicle (AV) systems.

Your primary responsibility will be to conduct thorough performance analysis and optimization of these networks, ensuring they operate efficiently both onboard the vehicle and during training on large-scale data centers. This position requires a deep understanding of software architecture, system performance, and latency issues, as you will be tackling various challenges that arise in these areas. You will collaborate with a team of talented engineers and researchers to develop solutions that improve the overall efficiency and reliability of our self-driving technology. Your work will directly contribute to making transportation safer and more accessible. The role demands a strong analytical mindset, particularly in performance troubleshooting, where you will utilize techniques such as profiling and the roofline model to identify bottlenecks and optimize performance. In addition to your technical skills, you will need to be adaptable and quick to learn new technologies, as the field of deep learning and autonomous systems is rapidly evolving. Strong communication skills are essential, as you will be working in a fast-paced environment with large code bases and collaborating with cross-functional teams.

In this role you will

  • Conduct performance analysis and optimization of Deep Learning networks running on the Autonomous Vehicle (AV).
  • Optimize software architecture, system performance, and latency for deep learning applications.
  • Work on deployment of deep learning models on the AV and training on large-scale data centers.
  • Troubleshoot performance issues using profiling and roofline model techniques.
  • Collaborate with cross-functional teams to enhance the efficiency of self-driving technology.

Required Qualifications

  • Minimum 5+ years of professional experience in software engineering.
  • BS, MS, or PhD in Computer Science or a related field.
  • Strong programming skills in CUDA, C++ and Python
  • Extensive experience in high-performance computing and parallel programming, specializing in optimizing workloads to reduce GPU memory usage, minimize latency, and/or maximize throughput.
  • Proficiency in leveraging performance analysis tools such as NVIDIA Nsight Systems , Nsight Compute and applying techniques like roofline model for performance optimization.
  • Hands-on experience in optimizing DL/ML workloads at the framework level using at least one deep learning framework (e.g., PyTorch, TensorFlow), ensuring efficient and scalable model deployment.
  • Strong understanding of the fundamentals of computer vision and transformer-based deep learning architectures, with proficiency in foundational neural network building blocks.
  • Strong analytical skills for diagnosing and troubleshooting performance bottlenecks in complex systems.
  • Demonstrated ability to quickly learn and adapt to emerging technologies and tools in a fast-paced environment
  • Experience working on large code bases in a fast-growing environment.
  • Strong communication skills, enabling effective teamwork across multidisciplinary teams.
  • Comfortable working in Linux/Unix environments.

Desirable Qualifications

  • Hands-on experience in motion planning or related fields such as robotics, autonomous systems, systems software, or computer vision.
  • Experience with TensorRT, OpenAI Triton, Mojo and other inference acceleration tools.

For roles based in Mountain View, CA and San Francisco, CA: The salary range for this position is $189,000 - $274,000 per year.

For roles based in Pittsburgh, PA: The salary range for this position is $171,000 - $247,000 per year.

Aurora's pay ranges are determined by role, level, and location. Within the range, the successful candidate's starting base pay will be determined based on factors including job-related skills, experience, qualifications, relevant education or training, and market conditions. These ranges may be modified in the future. The successful candidate will also be eligible for an annual bonus, equity compensation, and benefits.

  • Senior

Working at Aurora

At Aurora, we bring together extraordinarily talented and experienced people united by the strength of our values. We operate with integrity, set outrageous goals, and build a culture where we win together — all without any jerks.

We believe in-person work increases collaboration, empathy and our ability to lead effectively. As a result, we operate in a hybrid work environment where Aurorans are in office at least 3 days per week.

Our Careers page provides insight into what it is like to work at Aurora, and you can find all the latest updates in our Newsroom.

Our commitment to safety

At the core of everything we do is our commitment to safety. Building best-in-class self-driving technology will take time, and we believe that each employee at Aurora has a role in contributing to safety, every step of the way. Aurora expects commitment to our safety policies from every employee, and seeks candidates who take an active responsibility, can contribute to building an atmosphere of trust, and invest in the organization’s long-term success by prioritizing working safely, no matter what.

Our commitment to inclusion

Aurora considers candidates without regard to their race, color, religion, national origin, age, sex, gender, gender identity, gender expression, sexual orientation, marital status, pregnancy status, parent or caregiver status, ancestry, political affiliation, veteran and/or military status, physical or mental disability, or any other status protected by federal or state law. Aurora considers qualified applicants with criminal histories, consistent with applicable federal, state, and local law. We are also committed to providing reasonable accommodations for qualified individuals with disabilities and disabled veterans in our job application procedures. If you need assistance or an accommodation due to a disability, you may contact us at View email address on ev.careers.

For California applicants, information collected and processed as part of your application and any job applications you choose to submit is subject to Aurora’s California Employment Privacy Policy.

Vacancy posted 5 days ago
Similar jobs that could be interesting for youBased on the Staff Software Engineer, Deep Learning Acceleration in California vacancy
  • $190k - $300k

     ...at the forefront of software and hardware innovation...  ...remote.The Role: Staff Software Engineer, SIMD KernelsWhat you...  ...software, and machine learning fundamentals.Proficient...  ...FPGAs, DSPs, GPUs, AI accelerators using libraries such...  ...etc.Experience with a deep learning framework (... 
    Suggested
    Work experience placement
    Remote work

    d-Matrix

    Santa Clara, CA
    1 day ago
  • $130k - $220k

    Santa Clara, CASoftware Engineering - Motion Planning /...  ...-based virtual driver software for factory-built autonomous...  ...working with Plus to accelerate the deployment of next...  ...robotics and machine learning techniques from...  ...learning techniques such as deep learning, imitation... 
    Suggested
    Full time

    Plus.ai

    Santa Clara, CA
    18 hours ago
  • $193.93k - $352.29k

     ...components in machine learning model life cycle, to push...  .... Work with autonomy engineers to optimize, validate,...  ...robust, high quality software to increase our...  ...CUDA.Working experience deep learning frameworks (like...  ...are passionate about accelerating the benefits of robotics... 
    Suggested
    Work experience placement
    Immediate start
    Flexible hours

    Nuro

    Mountain View, CA
    4 days ago
  • $189.3k - $290.7k

    Description Role As a Staff Software Engineer on the Validation Foundations and Acceleration team within the Autonomous Vehicle (AV) organization, you will play...  ...–so you can focus on realizing your ambitions. Learn how GM supports a rewarding career that rewards... 
    Suggested
    Full time
    Local area
    Work from home
    Flexible hours

    General Motors

    Sunnyvale, CA
    3 days ago
  • $130k - $170k

     ...Description We are seeking a Staff Software Engineer to lead the development of...  ...and using agentic AI to accelerate development. This team...  ..., test and deployment. ~ Deep understanding of ML practices...  ...and some fine-tuning, deep learning, and training/algorithm... 
    Suggested
    Full time
    Local area
    Remote work

    NBCUniversal

    Universal City, CA
    17 days ago
  • $262k - $364k

     ...defining the architecture and technical road map for the software stack on our accelerator platforms.Drive large-scale technical programs from...  ...programming.Preferred qualifications:Master’s degree or PhD in Engineering, Computer Science, or a related technical field.8 years... 
    Worldwide

    Google

    Sunnyvale, CA
    2 days ago
  •  ...build great products that accelerate next-generation...  ...looking for an influential software engineer who is passionate...  ...technology.  THE PERSON:As a Staff Software Developer,...  ...this role requires a deep passion for software engineering...  ...) and Reinforcement Learning (e.g., RLHF, GRPO).... 

    AMD

    Santa Clara, CA
    3 days ago
  •  ...is responsible for developing machine learning (ML) algorithms that learn and...  ...Perception, Simulation, and Systems Engineering teams to accelerate our ability to validate our driving performance...  ...will be responsible for developing deep learned models that produce... 
    Temporary work
    Relocation package

    Zoox

    Foster, CA
    14 days ago
  • $187.18k - $201.22k

     ...orchestration with intelligent machines. Position Overview: The Staff Software Engineer plays a key role for the Software Engineering Applications...  ...integration of state-of-the-art autonomy features (deep learning models and pipelines, state-of-the-art mapping, perception... 
    Full time
    Work experience placement
    Work at office
    Immediate start
    Worldwide
    Flexible hours
    Night shift

    Brain Corp

    San Diego, CA
    1 day ago
  • $155k - $269k

     ...in AI, automotive, logistics, and deep tech. With offices in Toronto,...  ...the world in a positive way. To learn more visit: As a Graphics Software Engineer, you will create the next generation...  ...systems. - Experience with accelerated computing like CUDA, OpenCL, and... 
    Full time
    Work at office
    Work from home
    Flexible hours

    Waabi

    San Francisco, CA
    22 days ago
  • $245k - $350k

     ...list of the world’s top 100 private deep tech companies, ranking #12 globally...  ...November 2024.ResponsibilityAs the Staff/Senior Staff Software Engineer - Perception, you will be responsible...  ...Manage the team with world-class deep learning engineers and software engineers,... 
    Full time
    Temporary work

    Pony.ai

    Fremont, CA
    4 days ago
  • $189k - $303k

     ...Staff Machine Learning Engineer - Online Maps Aurora's mission is to deliver the benefits of self-driving...  ...and develop state-of-the art deep learning/machine learning models to improve...  ...Required qualifications: Excellent software engineering skills in Python and/or... 
    Work at office
    Local area
    3 days per week

    Aurora CO

    Mountain View, CA
    3 days ago
  • $141k - $242k

     ...Senior Or Staff Software Engineer The Mapping team at Waabi develops the foundational spatial intelligence of our autonomous fleet. As a...  ...renowned engineers and scientists specializing in robotics, deep learning, computer vision, and autonomous driving. We value... 
    Full time
    Work at office
    Work from home
    Flexible hours

    Waabi

    San Francisco, CA
    3 days ago
  • $160k - $200k

     ...cost and performance, while enabling accelerated growth for a diverse set of high-growth...  ...for an experienced Machine Control Software Engineer who can lead the design, development,...  ...systems (Cognex VisionPro, In-Sight Deep Learning). Build and debug state machines for... 
    Temporary work

    Halo Industries, Inc.

    Santa Clara, CA
    a month ago
  • $177k - $265.6k

     ...spirit to our collaborative teams. As a Staff Data Software Engineer (Level 5) located in Manhattan...  ...engagement opportunities such as lunch-and-learns, career development events, and...  ...in domains such as machine learning, deep learning, or large-scale data pipelines... 
    Full time
    Work experience placement
    Relocation package
    Shift work

    Northrop Grumman

    Manhattan Beach, CA
    1 day ago
  • $193.93k - $352.29k

     ...About the RoleWe are looking for a Senior/Staff Software Engineer to serve as a technical leader for...  ...critical intersection of Autonomy, Machine Learning, and Infrastructure, acting as an...  ...the following areas: Machine Learning, Deep Learning, Robotics , and have some... 
    Immediate start
    Flexible hours
    Shift work

    Nuro

    Mountain View, CA
    1 day ago
  • $251k - $310k

     ...tens of billions in simulation across 15+ U.S. states. Software Engineering builds the brains of Waymo's fully autonomous driving technology...  ...in areas like robotics, perception, decision-making and deep learning, while collaborating with hardware and systems engineers.... 
    Full time
    Remote work

    Waymo

    California
    8 days ago
  •  ...and optimize scalable machine learning infrastructure for large...  ...pipelines. Collaborate with engineering, data infrastructure, data analytics...  ...fields. ~ Experience with deep learning frameworks such as...  ...in machine learning and software engineering research publications... 
    Full time
    Work at office
    Remote work
    Flexible hours

    ServiceNow

    Mountain View, CA
    15 days ago
  •  ...understanding the changing Earth.   About the Job As a Staff Software Engineer for Computer Vision, you will develop and deploy across...  ..., radar algorithms, or SAR concepts Familiarity with deep learning approaches to 3D vision (e.g. NeRF, 3D Gaussian Splatting,... 
    Permanent employment
    Full time
    Remote work
    Night shift

    Array Labs

    Redwood City, CA
    22 days ago
  • $189k - $303k

     ...accessible for all. We’re searching for a Staff Machine Learning Engineer - Online Maps to join Aurora’s Maps...  ...and develop state-of-the art deep learning/machine learning models to improve...  ...Required Qualifications Excellent software engineering skills in Python and/or... 
    Work at office
    Local area
    3 days per week

    Aurora

    California
    13 days ago
  • $141k - $242k

     ...leaders in AI, automotive, logistics, and deep tech. With offices in Toronto, San...  ...impact the world in a positive way. To learn more visit: The Localization team at...  ...vehicle right now?  As a Senior or Staff Software Engineer on the Localization team, you will be a... 
    Full time
    Work at office
    Work from home
    Flexible hours

    Waabi

    San Francisco, CA
    22 days ago
  • $228k - $285k

     ...generations. Role Summary As a Staff Software Engineer, ML training and inference...  ...develops advanced machine learning algorithms that directly...  ...for candidates with deep knowledge and strong enthusiasm...  ...transformer architecture and ways to accelerate the training and inference... 
    Full time
    Contract work
    Local area

    Rivian

    California
    5 days ago
  • $148k - $260k

     ...in AI, automotive, logistics, and deep tech. With offices in Toronto,...  ...impact the world in a positive way. To learn more visit: As a Software Engineer in High-Performance Onboard...  ...architectures (e.g., CPU, GPU, specialized accelerators) and low-level system programming.... 
    Full time
    Work at office
    Work from home
    Flexible hours

    Waabi

    San Francisco, CA
    28 days ago
  • $189k - $303k

     ...accessible for all. We’re searching for a Staff Software Engineer and Machine Learning practitioner to join the...  ...actions and attributes, and developing deep semantic understanding of diverse...  ...-performance GPU programming for accelerated computing.Publication record at... 
    Work at office
    Local area
    3 days per week

    Aurora Innovation

    Mountain View, CA
    2 days ago
  •  ...Description   We are looking for general or deep learning compiler engineers to join our compiler team at Baidu’s...  ..., we advance compiler technology to accelerate applications in deep learning...  ..., infrastructure compiler and the software stack for our proprietary high-... 
    Full time
    Work at office
    Remote work
    Afternoon shift

    Baidu Usa

    Sunnyvale, CA
    1 day ago
  • $144k - $234k

     ...essential to our continued success. Our deep curiosity to learn and our eagerness to share diverse...  ...LL DO We’re looking for a Senior Software Engineer to join our Content Cards team,...  ...strengthen platform foundations that accelerate future innovation across in-product... 
    Full time
    Work at office
    Local area
    Flexible hours

    Braze

    San Francisco, CA
    1 day ago
  •  ...About the Team The Kernels team at OpenAI builds the low-level software that accelerates our most ambitious AI research. We work at the boundary of hardware and software, developing high-performance kernels, distributed system optimizations, and runtime improvements... 
    Full time

    OpenAI

    San Francisco, CA
    1 day ago
  •  ...talented minds in research, engineering, and operations to create products...  ...the Role The Engineering Acceleration-focused engineer on the...  ..., test, ship, and debug software across device and cloud surfaces...  ...for an engineer with deep experience in developer productivity... 
    Full time
    Work at office
    Relocation package

    OpenAI

    San Francisco, CA
    1 day ago
  • $120k - $140k

     ..., AI-enabled autonomy software suite that enables us...  ...Lodestar, as a   Software Engineer - Localization, State...  ...with machine learning to track targets, predict...  ...inference) ~ Experience with deep learning frameworks (...  ...Proficiency with GPU acceleration (CUDA, TensorRT) for... 
    Permanent employment
    Full time
    Flexible hours

    Lodestar Corporation

    San Francisco, CA
    1 day ago
  •  ...have an opening to hire a Staff Software Engineer - Agent OptimizationTemporal...  ...AI Foundations team is to accelerate Temporal adoption across the...  ...ecosystem. Our approach combines a deep understanding of use cases...  ...systems using both machine learning and systems... 
    Work experience placement

    Temporal Technologies

    San Francisco, CA
    5 days ago

Do you want to receive more vacancies?

Subscribe and receive similar vacancies to Staff Software Engineer, Deep Learning Acceleration. Be the first to apply!