Posted Aug 16

Nvidia is hiring a
Senior Software Engineer - AI

US, CA, Santa Clara • US, CA, Santa Clara • US, CA, Santa Clara
Full time

NVIDIA is seeking software engineers for its Deep Learning Algorithms team to develop and commercialize Artificial Intelligence solutions to diverse real-world problems related to Generative AI (Large Language Models, Multimodal), NLP, CV, Speech, Text, Recommendation Systems, Graph Neural Networks (GNN) etc.

In this role you will be crafting and implementing new end-to-end AI solutions, define and influence existing APIs, tuning performance, analyzing functionality coverage across multiple solutions, building larger and coherent AI toolsets. You will be interacting with internal partners, users, and members of the open source community to analyze, define and implement highly optimized solutions. If the idea of pushing the boundaries of state-of-the-art research and development excites you? Are interested in getting exposure to the entire DL SW stack? Then come join the team and build GPU-accelerated DL platform used worldwide.

What you’ll be doing:

  • Research, prototype and develop effective tools and infrastructure pipelines

  • Tackle end to end AI training and inference-deployment problems (data fetching, pre-processing, orchestrating the training runs, model serving)

  • Work at the intersection of deep learning applications, libraries, frameworks, and the entire software stack.

  • Analyze, influence and improve deep learning libraries, frameworks and APIs according to good engineering practices

  • Collaborate with team members and other teams

What we need to see:

  • An MS and at least 3+ years of relevant industry experience in software development or a BS (or equivalent experience) and 5+ years of experience.

  • Excellent Python and C/C++ programming knowledge

  • Expertise in standard software engineering methodology, e.g. design patterns, data structure and algorithms, unit testing, test automation, continuous integration, code reviews, design documentation, debugging/profiling and version control

  • Strong analytical skill, Good communication, Strong time-management and organization skills for coordinating multiple initiatives, priorities and implementations of new technology and products into very complex projects.

  • Strong knowledge of deep learning fundamentals and techniques.

Ways to stand out from the crowd:

  • Knowledge of CPU and/or GPU architecture

  • Prior experience with large scale AI training and understanding of the compute system concepts (latency/throughput bottlenecks, pipelining, multiprocessing etc)

  • Prior background working at the intersection of deep learning applications, libraries, frameworks and the software stack.

  • Hands-on experience with one of the deep learning frameworks (e.g. JAX, PyTorch, TensorFlow, Keras, MxNet), or deep neural networks inference and deployment environments would be an asset (e.g. TRT, ONNX, Triton)

  • Experience with cloud computing (e.g. end-to-end pipelines for AI training and inference on CSP (AWS/Azure/GCP))

  • Relevant working experience with Docker and Kubernetes is a big plus.

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 on the planet working with us. If you're creative and autonomous, we want to hear from you!

The base salary range is $144,000 - $224,250. 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 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.

Please mention that you found the job on ARVR OK. Thanks.