Current jobs related to Senior Performance Software Engineer, Deep Learning Libraries - Santa Clara CA United States - NVIDIA


  • Santa Clara, California, United States NVIDIA Full time

    Senior Performance Software Engineer for Deep Learning LibrariesWe are seeking a highly skilled Senior Performance Software Engineer to join our team at NVIDIA. As a key member of our Deep Learning Library team, you will be responsible for developing optimized code to accelerate linear algebra and deep learning operations on NVIDIA GPUs.Key...


  • Santa Clara, United States NVIDIA Full time

    Senior Performance Engineer - Deep Learning NVIDIA is hiring senior software engineers to build and optimize the tools Deep Learning engineers use across the world to design, develop, and deploy AI applications. We are an ambitious, forward-thinking and diverse team that influences all areas of NVIDIA's AI platform and directly contributes to premiere Deep...


  • Santa Clara, California, United States NVIDIA Full time

    About NVIDIANVIDIA is a leader in the field of artificial intelligence and deep learning. We're committed to fostering a diverse work environment and proud to be an equal opportunity employer.Job SummaryWe're seeking a highly skilled Senior Deep Learning Software Engineer to join our cuDNN team. As a key member of our team, you'll be responsible for...


  • US, CA, Santa Clara NVIDIA Full time

    NVIDIA is hiring senior software engineers to build and optimize the tools Deep Learning engineers use across the world to design, develop, and deploy AI applications. We are an ambitious, forward-thinking and diverse team that influences all areas of NVIDIA's AI platform and directly contributes to premiere Deep Learning frameworks - PyTorch, JAX and...


  • US, CA, Santa Clara NVIDIA Full time

    We're now looking for a Senior Deep Learning Software Engineer for our cuDNN team!Do you love writing fast code and crafting software systems to solve complex problems? We are looking for hardworking software engineers to help design, build, and ship cuDNN: our GPU-accelerated library of primitives for deep neural networks. Intelligent machines powered by AI...


  • US, CA, Santa Clara NVIDIA Full time

    NVIDIA is hiring software engineers at all experience levels to build and optimize the tools Deep Learning engineers use across the world to design, develop, and deploy AI applications. This position will embed you in an ambitious and diverse team that influences all areas of NVIDIA's AI platform and directly contributes to premiere Deep Learning frameworks...


  • Santa Clara, California, United States NVIDIA Full time

    About NVIDIANVIDIA is a leader in the field of artificial intelligence and deep learning, and we're seeking a talented Senior Performance Engineer to join our team.Job SummaryWe're looking for a highly skilled engineer to develop and optimize open-source libraries, study and tune deep learning training workloads, and build and support NVIDIA submissions to...


  • Santa Clara, California, United States NVIDIA Full time

    About the RoleWe are seeking a highly skilled Senior Deep Learning Algorithm Engineer to join our team at NVIDIA. As a key member of our team, you will play a critical role in optimizing deep learning training workloads on state-of-the-art hardware and software platforms.Key ResponsibilitiesUnderstand, analyze, profile, and optimize deep learning training...


  • Santa Clara, California, United States Blockchain Technologies. LLC Full time

    Job OverviewBlockchain Technologies, LLC is seeking a highly skilled Machine Learning Engineer to join our team. As a Machine Learning Engineer, you will be responsible for developing and implementing deep learning models using TensorFlow, PyTorch, and Keras.Key Responsibilities:Design and develop deep learning models for computer vision applications,...


  • Santa Clara, United States Techions Full time

    About Techions:Techions is not just a company; it's a dynamic hub of innovation and creativity, where groundbreaking ideas come to life. With a relentless pursuit of excellence, we are revolutionizing the tech industry one solution at a time. Our team of visionaries is passionate about pushing the boundaries of what's possible and delivering unparalleled...


  • Santa Clara, United States Techions Full time

    About Techions:Techions is not just a company; it's a dynamic hub of innovation and creativity, where groundbreaking ideas come to life. With a relentless pursuit of excellence, we are revolutionizing the tech industry one solution at a time. Our team of visionaries is passionate about pushing the boundaries of what's possible and delivering unparalleled...


  • Santa Clara, California, United States NVIDIA Corporation Full time

    Job DescriptionNVIDIA Corporation is seeking a highly skilled Senior Distributed Machine Learning Engineer to join our team focused on developing cutting-edge GPU-accelerated distributed machine learning solutions.About the RoleDesign and develop new user-friendly APIs and libraries to optimize the use of existing deep learning and machine learning...


  • Santa Clara, California, United States NVIDIA Full time

    About the RoleWe are seeking a highly skilled Senior System Software Engineer to join our team working on the CUDA driver and runtime, core components of our platform for accelerating general purpose computation on the GPU.Key ResponsibilitiesInvestigate complex performance problems and deliver robust solutions that accelerate applications.Evangelize,...


  • Santa Clara, California, United States Aitopics Full time

    Job DescriptionAitopics is seeking a highly skilled Senior Software Quality Assurance Engineer to join our team. As a Senior Software Quality Assurance Engineer, you will be responsible for ensuring the quality and reliability of our Deep Learning software.Key ResponsibilitiesWork closely with cross-functional teams to understand test requirements and take...


  • Santa Clara, California, United States AMD Full time

    JOIN AMD AND MAKE A DIFFERENCEAt AMD, we are dedicated to revolutionizing lives through our advanced technology, enhancing our industry, communities, and the global landscape. Our vision is to create exceptional products that propel next-generation computing experiences, serving as the foundation for data centers, artificial intelligence, personal computing,...


  • Santa Clara, California, United States Robotics Prcocess Automation, LLC Full time

    Job OverviewJob ID: J36993Specialized Area: Machine LearningJob Title: Machine Learning EngineerDuration: 10 MonthsEmployment Type: W-2 (Consultant must be on our company payroll. C2C is not allowed)We are seeking a highly skilled Software Engineer to join our team at Robotics Process Automation, LLC and contribute to the development of a Deep Learning...


  • Santa Clara, United States NVIDIA Full time

    We are hiring senior engineers to work on the CUDA driver and runtime, core components of our platform for accelerating general purpose computation on the GPU. Our team analyzes performance of applications, investigates bottlenecks in software or hardware and delivers features and improvements to better realize the potential of NVIDIA hardware for a growing...


  • Santa Clara, United States AIxMed, Inc. Full time

    Job Title: Senior Software Engineers – AI/Computer Vision: Computational CytopathologySummary:Embark on a journey at the cutting edge of digital cytopathology with AIxMed, a pioneering startup leveraging AI and computer vision to transform digital cytology. In the role of Senior Software Engineer, you will be instrumental in crafting innovative software...


  • Santa Clara, United States AIxMed, Inc. Full time

    Job Title: Senior Software Engineers – AI/Computer Vision: Computational CytopathologySummary:Embark on a journey at the cutting edge of digital cytopathology with AIxMed, a pioneering startup leveraging AI and computer vision to transform digital cytology. In the role of Senior Software Engineer, you will be instrumental in crafting innovative software...


  • US, CA, Santa Clara NVIDIA Full time

    We are hiring senior engineers to work on the CUDA driver and runtime, core components of our platform for accelerating general purpose computation on the GPU. Our team analyzes performance of applications, investigates bottlenecks in software or hardware and delivers features and improvements to better realize the potential of NVIDIA hardware for a growing...

Senior Performance Software Engineer, Deep Learning Libraries

3 months ago


Santa Clara CA United States NVIDIA Full time
We are now looking for a Senior Performance Software Engineer for Deep Learning Libraries Do you enjoy tuning parallel algorithms and analyzing their performance? If so, we want to hear from you As a deep learning library performance software engineer, you will be developing optimized code to accelerate linear algebra and deep learning operations on NVIDIA GPUs. The team delivers high-performance code to NVIDIA's cuDNN, cuBLAS, and TensorRTlibraries to accelerate deep learning models. The team is proud to play an integral part in enabling the breakthroughs in domains such as image classification, speech recognition, and natural language processing. Join the team that is building the underlying software used across the world to power the revolution in artificial intelligence We're always striving for peak GPU efficiency on current and future-generation GPUs. To get a sense of the code we write, check out our CUTLASS open-source project showcasing performant matrix multiply on NVIDIA's Tensor Cores with CUDA. This specific position primarily deals with code lower in the deep learning software stack, right down to the GPU HW.

What you'll be doing:

Writing highly tuned compute kernels, mostly in C++ CUDA, to perform core deep learning operations (e.g. matrix multiplies, convolutions, normalizations)

Following general software engineering best practices including support for regression testing and CI/CD flows

Collaborating with teams across NVIDIA:

CUDA compiler team on generating optimal assembly code

Deep learning training and inference performance teams on which layers require optimization

Hardware and architecture teams on the programming model for new deep learning hardware features

What we need to see:

Masters or PhD degree or equivalent experience in Computer Science, Computer Engineering, Applied Math, or related field

6+ years of relevant industry experience

Demonstrated strong C++ programming and software design skills, including debugging, performance analysis, and test design

Experience with performance-oriented parallel programming, even if it's not on GPUs (e.g. with OpenMP or pthreads)

Solid understanding of computer architecture and some experience with assembly programming

Ways to stand out from the crowd:

Tuning BLAS or deep learning library kernel code

CUDA/OpenCL GPU programming

Numerical methods and linear algebra

LLVM, TVM tensor expressions, or TensorFlow MLIR

NVIDIA is widely considered to be one of the technology world's most desirable employers. We have some of the most forward-thinking and hard working people in the world working for us. If you're creative, autonomous, and love a challenge, consider joining our Deep Learning Library team and help us build the real-time, cost-effective computing platform driving our success in this exciting and quickly growing field.

The base salary range is 180,000 USD - 339,250 USD. Your base salary will be determined based on your location, experience, and the pay of employees in similar positions.

You will also be eligible for equity and benefits. NVIDIA accepts applications on an ongoing basis.

NVIDIA is committed to fostering a diverse work environment and proud to be an equal opportunity employer. As we highly value diversity in our current and future employees, we do not discriminate (including in our hiring and promotion practices) on the basis of race, religion, color, national origin, gender, gender expression, sexual orientation, age, marital status, veteran status, disability status or any other characteristic protected by law.