Senior Software Engineer, AI

Key Skills

Aartificial intelligence, Analytical and Problem solving, C++, Debugging, Deep Learning, Python Programming, PyTorch, TensorRT

Job Description

NVIDIA's high-performance computing platforms are powering the AI revolution! Our GPUs deliver industry-leading performance on many applications including generative AI through our impressive suite of software products like TensorRT and cuDNN. Come join our team and develop software that integrates into many of these cutting-edge AI products. If you are excited to learn how mathematical operations in large language models are implemented, then apply today! Become part of the team that brings new GPU technologies for AI across NVIDIA's software stack, and be among the first to breathe life into new silicon.

What you'll be doing:

• As a member of our team, you will use your design abilities, coding expertise, and creativity to deliver a GPU-accelerated library used in NVIDIA products ranging from Data Center, Automotive and Embedded systems.

• Architect and implement new features to deliver an amazing user experience for our internal customers, like cuDNN, cuBLAS, and TensorRT

• Design and develop validation methodologies so that our library is robust across a diverse cross product of hardware, OSes, and platforms, including simulators and emulators for future GPUs

• Write effective, maintainable, and well-tested code

What we need to see:

• MS or PhD degree in Computer Science, Electrical Engineering or related field (or equivalent experience)

• Deep C++ programming and problem-solving skills, including debugging, performance analysis, documentation, and test design

• 6+ years of relevant software development experience

• Background in working with large codebases

• The ability to work independently, define project goals and scope, and lead your own development effort

• Strong written communication as well as presentation skills

Ways to stand out from the crowd:

• Prior experience with parallel computing

• Understanding of Deep Learning and AI

• Fluency in build systems (e.g. Make, CMake, and/or Bazel)

• Multiple contributions to open source software

NVIDIA is widely considered to be one of the technology world’s most desirable employers. We have some of the most forward-thinking and hardworking people in the world working for us. If you're creative and autonomous, we want to hear from you!

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.