Senior Deep Learning Software Engineer

We are now looking for a Senior Deep Learning Software Development Engineer! Academic and commercial groups around the world are using GPUs to power a revolution in deep learning, enabling breakthroughs in problems from image classification to speech recognition and natural language processing. By tapping into the unlimited potential of AI to define the next era of computing, an era in which our GPU acts as the brains of computers, robots, and self-driving cars that can understand the world, we're doing whatever it takes to include vision, innovation, with the world’s best talent. Join us in building software which will be used by the entire world while collaborating directly with deep learning framework developers working on code bases that are shared with the community.

As an experienced and skilled engineer, your collaboration in developing new technology will have a global impact by building software with applications ranging from fraud detection, recommendation systems to material science, and beyond. The ideal candidate should be able to work effectively in a fast-paced team that prioritizes customer focus. Strong communication skills are essential for this role in collaborating with an ambitious group of people pushing the boundaries of AI and machine learning.

What you'll be doing:

In this role, you will be responsible for developing core deep learning algorithms for both internal and 3rd party codebases. As part of our team, you will be encouraged to actively participate in the software development community and make valuable contributions to well-known software packages such as PyG, PyTorch, Modulus etc. The scope of these efforts ranges from integrating the latest CUDA libraries and features, performance tuning and analysis, implementing new algorithms, and other general software engineering work.

What we need to see:

• Masters or PhD or equivalent experience in Computer Science, Artificial Intelligence, Applied Math, or related field.

• Capability to work autonomously, establish project objectives and boundaries, engage directly with the development community, and coordinate your own development endeavor.

• Excellent C++ programming and software design skills, including debugging, performance analysis, and test design. Python experience also highly desired.

• Distributed systems programming or numerical methods.

• GPU programming experience (CUDA or OpenCL) desired but not required

• Prior experience with machine learning algorithms and frameworks. Especially experience frameworks such as PyTorch, JAX, DGL, PyG etc.

• Experience contributing to or managing a significant project involving community collaboration - proficiency in using online platforms for code sharing, bug tracking, code merging, licensing challenges, patch management, etc.

• 7+ years of relevant software development experience

• Occasional travel to conferences and for customer visits may be required.

Ways to stand out from the crowd:

• Good at CUDA programming and parallel computing.

• Advanced degree or equivalent experience in Computer Science, Electrical Engineering, or related field.

• Experience contributing to or managing open-source projects.

NVIDIA is widely considered to be one of the technology world’s most desirable employers. We have some of the most dedicated individuals 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 - 276,000 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.