NVIDIA is seeking an engineering leader to manage the activities related to productizing Artificial Intelligence (AI)/Deep Learning(DL) software and models. If the idea of pushing the boundaries of state-of-the-art research and development excites you, and you are interested in getting exposure to the entire AI software stack, come join us to build GPU-accelerated AI platform.
If you have leadership experience with good understanding of AI frameworks, libraries and algorithms, then this role may be a great fit for you!. In this role you will lead, mentor forward-thinking engineers and will be responsible for ownership of related activities and interactions with teams across NVIDIA. You will be working with key internal partners on priority alignment across relevant teams for roadmap development of highly optimized novel and state-of-the-art AI software systems and algorithms.
What you’ll be doing:
- Plan, schedule, and lead the execution of projects and activities of the team related to NeMo Framework and Megatron Core
- Solve large-scale, end-to-end AI training and inference-deployment challenges
- Performance tuning and optimizations of AI/DL application framework & software components.
- Collaborate with internal and external customers to align on priorities across business units and geographic locations
- Grow and develop a world-class team
What we need to see:
- Masters or PhD degree (or equivalent experience)
- At least 10 overall years of relevant industry experience in Computer Science, Artificial Intelligence, Applied Math, or related field
- 3+ years of demonstrated experience in leading and mentoring software engineering teams
- Working knowledge of one of the deep learning frameworks (e.g. PyTorch, JAX)
- Excellent understanding of SDLC practices including architecting, testing, continuous integration, and documentation
- Strong communication, analytical and organization skills
Ways to stand out from the crowd:
- Prior experience with large scale AI training and understanding of the compute system concepts (latency/throughput bottlenecks, pipelining, multiprocessing), related performance analysis and tuning.
- Familiarity with generative AI techniques applied to LLM and Multi modality learning (Image, Video, Speech).
- Knowledge of GPU/CPU architecture and related numerical software.
- Contributions to open source deep learning frameworks.
As NVIDIA makes inroads into the Datacenter business, our team plays a central role in getting the most out of our exponentially growing datacenter deployments as well as establishing a data-driven approach to hardware design and system software development. We collaborate with a broad cross section of teams at NVIDIA ranging from DL research teams to CUDA Kernel and DL Framework development teams, to Silicon Architecture Teams. As our team grows, and as we seek to identify and take advantage of long term opportunities, our skillset needs are expanding as well.
NVIDIA is widely considered to be one of the technology world’s most desirable employers. We have some of the most forward-thinking and hardworking people on the planet working for us. If you're creative and autonomous, we want to hear from you!
The base salary range is 220,000 USD - 419,750 USD. 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 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.
