Software Engineer, Deep Learning Cudnn - New College Grad 2024

Do you love writing fast code and crafting software systems to solve complex problems? We are looking for hardworking software engineers to help design, build, and ship cuDNN: our GPU-accelerated library of primitives for deep neural networks. Intelligent machines powered by AI computers that can learn, reason, and interact with people are no longer science fiction. This is truly an extraordinary time. The era of AI has begun, and we are powering it. If this role seems like a good match for your skills and interests, tell us why you think you might be a great fit for our team, and we'd love to tell you more about what we do!

What You’ll Be Doing

• Develop production-quality software that ships as part of NVIDIA's AI software stack, including optimized large language model (LLM) support.
• Adapt to the constantly evolving AI industry by being agile and excited to contribute across the codebase, including API design, software architecture, performance modeling, testing, and GPU kernel development.
• Analyze the performance of important workloads, tuning our current software, and proposing improvements for future software.
• Work with cross-collaborative teams of deep learning software engineers and GPU architects to innovate across applications like generative AI, autonomous driving, computer vision, and recommender systems.

What We Need To See

• B.S. degree or higher in computer science (or similar) or equivalent experience
• Experience with performance analysis, profiling, and code optimization.
• Strong programming skills in C/C++ development and familiarity with Python.
• Good understanding of linear algebra.
• Excellent problem solving skills, including applications of algorithms and data structures.
• Familiarity with the latest trends in machine learning.

Ways To Stand Out From The Crowd

• Strong experience with data science, statistical analysis, and visualization.
• Experience with computer architecture and building performance models for CPUs, GPUs, or other accelerators.
• Practical experience with machine learning, especially deep learning.
• Previous work on large complex codebases with many other developers, especially libraries, compilers, or system software.
• GPU programming and optimization expertise (e.g. CUDA or OpenCL).

NVIDIA is widely considered to be one of the technology world’s most desirable employers with some of the most forward-thinking, inventive, and hardworking people working here - we need your help to build our future. Are you creative, ambitious, and love challenges? We want to hear from you!

The base salary range is $100,000 - $184,000. 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.