Senior Performance Engineer - Deep Learning
NVIDIA
California, United States
See who NVIDIA has hired for this role
Pay found in job post
Retrieved from the description.
Base pay range
$180,000.00/yr - $339,250.00/yr
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
You will also be eligible for equity and benefits . NVIDIA accepts applications on an ongoing basis.
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.#deeplearning
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.
- 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.
- 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.
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.#deeplearning
-
Seniority level
Mid-Senior level -
Employment type
Full-time -
Job function
Engineering -
Industries
Computer Hardware Manufacturing, Software Development, and Computers and Electronics Manufacturing
Referrals increase your chances of interviewing at NVIDIA by 2x
See who you knowGet notified about new Senior Performance Engineer jobs in California, United States.
Sign in to create job alertSimilar jobs
People also viewed
-
Software Engineer L5, LLM Compute & Serving Systems
Software Engineer L5, LLM Compute & Serving Systems
-
Software Engineer - Remote (Global), Full-Time at Mixrank
Software Engineer - Remote (Global), Full-Time at Mixrank
-
Junior/Entry Software Engineer
Junior/Entry Software Engineer
-
Software Engineer – Full Stack
Software Engineer – Full Stack
-
Software Developer(REMOTE)
Software Developer(REMOTE)
-
Software Developer (Remote)
Software Developer (Remote)
-
Junior Back End Engineer
Junior Back End Engineer
-
Software Developer - Junior
Software Developer - Junior
-
New Graduate Software Engineer (Palo Alto)
New Graduate Software Engineer (Palo Alto)
-
Software Developer
Software Developer
Similar Searches
Explore collaborative articles
We’re unlocking community knowledge in a new way. Experts add insights directly into each article, started with the help of AI.
Explore More