Manager, Artificial Intelligence 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 224,000 USD - 425,500 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.