Deep Learning Software Engineer - Generative AI

NVIDIA is at the forefront of generative AI models, from language to images. Doing what’s never been done before takes vision, innovation, and the world’s best talent. Join the team building Deep Learning software which will be used by the entire world. Work with the scientific community to implement the latest AI solutions. In this role you will be interacting with internal partners, users, and members of the open source community to analyze, define and implement highly optimized DL algorithms and tools. The scope of these efforts includes a combination of implementing new algorithms, performance tuning and analysis, defining APIs, analyzing functionality coverage, and other general software engineering work. Ability to work in a dynamic, fast-paced environment is required as well as strong interpersonal skills.Are you ready for this challenge?What you’ll be doing:Develop tools and solutions for designing efficient hardware configurations for DL modelsSolve complex problems related to hardware management and profiling of Deep Learning workloadsAnalyze and improve performance of DL implementationsCollaborate with team members and other partnersWhat we need to see:MS degree in Computer Science, Artificial Intelligence, Computer Architecture or related technical field or equivalent experience3+ years of work experience in AI development (at least one DL Framework - preferably PyTorch)Excellent Python programming skillsMachine learning fundamentals (linear algebra, probability theory, optimization, supervised/unsupervised/self-supervised ML, etc.)Hands-on experience with Deep Learning (Convolutional Neural Networks, Transformers, Diffusion Models, etc.)Good understanding of hardware architecture (CPU, GPU, storage, networking)Willing to take action and have strong analytical skillsStrong time-management and organization skills for coordinating multiple initiatives, priorities and implementations of new technology and products into very complex projectsGood communication and documentation habitsWays to stand out from the crowd:PhD in Computer Science, Artificial Intelligence, Computer Architecture or related technical fieldExperience in design and implementation of complex systems with multiple dependenciesKnowledge of CPU (x86, ARM) and/or GPU architectureExperience with HPC clusters or/and clouds (AWS, Azure, GCP, OCI)Exposure to performance optimizationNVIDIA is widely considered to be one of the technology world’s most desirable employers. We have some of the most brilliant and forward-thinking people in the world working for us. If you're creative and autonomous, we want to hear from you! We are an equal opportunity employer and value diversity at our company. We do not discriminate on the basis of race, religion, color, national origin, gender, sexual orientation, age, marital status, veteran status, or disability status.