NVIDIA in California seeks a dedicated AI Software Engineer to enable and optimize open models using its advanced inference software stack. Ideal candidates should have a PhD in CS, EE, or equivalent, with at least 3 years of experience in deep learning and neural networks. Applicants will work on performance profiling and GPU optimization, contributing to open-source projects. The salary range is between 152,000 USD and 241,500 USD for Level 3, and 184,000 USD to 287,500 USD for Level 4, alongside equity and benefits.
Qualifications
PhD in a related field.
3+ years of experience in deep learning and neural networks.
Proficient with AI or HPC frameworks.
Responsibilities
Optimize open models on NVIDIA’s accelerated software stack.
Deliver production code to open‑source frameworks.
Analyze bottlenecks across the inference stack.
Skills
Deep learning
Neural networks
Performance profiling
GPU optimization
PyTorch
Computer architecture
Education
PhD in CS, EE, CSEE or equivalent
Tools
CUDA
OpenCL
Job description
NVIDIA in California seeks a dedicated AI Software Engineer to enable and optimize open models using its advanced inference software stack. Ideal candidates should have a PhD in CS, EE, or equivalent, with at least 3 years of experience in deep learning and neural networks. Applicants will work on performance profiling and GPU optimization, contributing to open-source projects. The salary range is between 152,000 USD and 241,500 USD for Level 3, and 184,000 USD to 287,500 USD for Level 4, alongside equity and benefits.