NVIDIA is hiring senior software engineers to build and optimize the tools Deep Learning engineers use across the world to design, develop, and deploy AI applications. We are an ambitious, forward-thinking and diverse team that influences all areas of NVIDIA's AI platform and directly contributes to premiere Deep Learning frameworks - PyTorch, JAX and TensorFlow. We work with multiple teams both inside and outside of NVIDIA across fields, as well as collaborate with the open-source community to optimize the best AI platform in the world!
What you will be doing:
Develop and optimize open-source libraries, like Transformer Engine, which enables the fastest training of Large Language Models using low precision data formats, and TensorFlow Distributed Embeddings, providing ability to easily scale training of huge recommender systems on multiple GPUs.
Study and tune Deep Learning training workloads at large scale, including important enterprise models.
Build and support NVIDIA submissions to community benchmarks like MLPerf.
Optimize the performance of influential, modern Deep Learning models coming out of academic and industry research, for NVIDIA GPUs and systems.
Explore new technologies and advise design of new hardware generations and core platform software components.
What we need to see:
BS in Computer Science, Electrical Engineering or a related field (or equivalent experience).
Demonstrated ability with 6+ years of C++ and Python programming.
Strong background with parallel programming, preferably on GPUs.
Knowledge of Computer Architecture and/or Operating Systems.
Proven experience developing large software projects.
Excellent verbal and written communication skills.
Ways to stand out from the crowd:
Experience with Deep Learning Frameworks, like PyTorch, JAX, Tensorflow or MXNet.
Experience training language models.
Background with performance analysis and profiling of workloads.
Participation in the open-source community.
Proven experience working with multidisciplinary teams.
You will also be eligible for equity and benefits. NVIDIA accepts applications on an ongoing basis.
Other Jobs from NVIDIA
Senior CPU Verification Engineer
Senior CPU Verification Engineer
Senior Verification Engineer - Memory Subsystem
Senior Verification Engineer - Memory Subsystem
Similar Jobs
Staff Machine Learning Engineer
Machine Learning Engineer
Staff Engineer - Machine Learning
Senior Verification Engineer - Memory Subsystem
Senior Verification Engineer - Memory Subsystem
Senior Verification Engineer, PCIE
There are more than 50,000 engineering jobs:
Subscribe to membership and unlock all jobs
Engineering Jobs
60,000+ jobs from 4,500+ well-funded companies
Updated Daily
New jobs are added every day as companies post them
Refined Search
Use filters like skill, location, etc to narrow results
Become a member
π₯³π₯³π₯³ 401 happy customers and counting...
Overall, over 80% of customers chose to renew their subscriptions after the initial sign-up.
To try it out
For active job seekers
For those who are passive looking
Cancel anytime
Frequently Asked Questions
- We prioritize job seekers as our customers, unlike bigger job sites, by charging a small fee to provide them with curated access to the best companies and up-to-date jobs. This focus allows us to deliver a more personalized and effective job search experience.
- We've got about 70,000 jobs from 5,000 vetted companies. No fake or sleazy jobs here!
- We aggregate jobs from 5,000+ companies' career pages, so you can be sure that you're getting the most up-to-date and relevant jobs.
- We're the only job board *for* software engineers, *by* software engineersβ¦ in case you needed a reminder! We add thousands of new jobs daily and offer powerful search filters just for you. π οΈ
- Every single hour! We add 2,000-3,000 new jobs daily, so you'll always have fresh opportunities. π
- Typically, job searches take 3-6 months. EchoJobs helps you spend more time applying and less time hunting. π―
- Check daily! We're always updating with new jobs. Set up job alerts for even quicker access. π
What Fellow Engineers Say