The PyTorch Team @ NVIDIA is hiring passionate parallel programmers. Join us to design and build the tools used by millions of AI practitioners deploying AI applications scalable to thousands of GPUs. Our team is responsible for the continual delivery of best in class experience on NVIDIA's hardware with PyTorch. Apply for an opportunity to collaborate with many multi-disciplinary engineering teams within NVIDIA and internationally in the PyTorch open source community to deliver our customers the best of NVIDIA software.In this position you will learn innovative techniques from NVIDIA's domain experts for efficiently programming the world's most sophisticated computers. Build these techniques into NVIDIA/Fuser (commonly known as "nvFuser") applying our groundbreaking Parallel Programming Theory, allowing these optimization techniques to be applied to algorithms broadly, automatically, and safely to algorithms written in Numpy and PyTorch. Beyond building nvFuser influence and improve the entire software stack all the way from users to the CUDA compiler, as well as influence the future design of NVIDIA's hardware platform. Join our ambitious and diverse team who strive to lead the best in AI programming.What you will be doing:Design and build core PyTorch components.Collaborate with NVIDIA’s hardware and software teams to improve GPU performance in PyTorch.Design, build and support production AI solutions used by enterprise customers and partners.Work with internal applied researchers to improve their AI tools.Advise design of new hardware generations.What we need to see:BS Computer Science, Computer Architecture, Electrical Engineering or a related field (or equivalent experience).6+ years of experience with C++ and Python programming.Demonstrated experience developing large software projects.We require excellent verbal and written communication skills.Contributions to PyTorch, Numpy, JAX, TensorFlow or similar system.The base salary range is $176,000 - $333,500. 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.
View Original Job Posting