Senior Software Engineer - Ai

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
• 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
• Tackle end to end AI training and inference-deployment problems (data fetching, pre-processing, orchestrating the training runs, model serving)
• Collaborate with team members and other teams

What We Need To See

• 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.
• 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

Ways To Stand Out From The Crowd

• 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))
• Prior background working at the intersection of deep learning applications, libraries, frameworks and the software stack.
• Relevant working experience with Docker and Kubernetes is a big plus.
• Prior experience with large scale AI training and understanding of the compute system concepts (latency/throughput bottlenecks, pipelining, multiprocessing etc)
• Knowledge of CPU and/or GPU architecture

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.