Nvidia Senior Deep Learning Software Engineer 

United States, California 

638594847

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.