Nvidia Senior Performance Engineer - AI Science 

United States, Texas 

625539282

What you'll be doing:

• Optimizing the computational performance of the latest innovations in AI that address specific, domain-relevant bottlenecks.

• Designing and implementing computationally performant features for large scale, CUDA-backed ML training and inference frameworks, and data science tooling.

• Developing and maintaining an HPC software stack for generative machine learning models in digital biology and beyond

• Collaborating with multiple HPC, AI infrastructure, and research teams

• Driving the testing and maintenance of the algorithms and software modules

What we need to see:

• Advanced degree in a quantitative field such as Computer Science, Computational Biology, Computational Chemistry, Physics, Mathematics, or equivalent experience

• 8+ years of relevant experience

• Proven track record in performance engineering as well as software design, building and packaging and launching software products

• Deep understanding of parallel programming in C++, Python; CUDA programming experience

• Proficient in modern machine learning frameworks such as PyTorch, TensorFlow, JAX, Warp

• Learn from and teach others to keep up to speed on the latest developments and tools in the field.

• Ability to work together in a tight-knit team environment.

Ways to stand out from the crowd:

• Interest and experience in the drug discovery domain

• Track record of code contributions that accelerate AI/ML bottlenecks

• Experience working with cutting edge methods in AI