Senior AI and Machine Learning Engineer

NVIDIA
Apply Now

Job Description

At NVIDIA, we are at the forefront of advancing the capabilities of artificial intelligence. We are seeking an ambitious and forward-thinking deep learning researcher to contribute to the development of next-generation optimizations aimed at efficiently serving generative AI at scale. In this role, you will be developing and exploring new cutting-edge AI optimization techniques, operating at the intersection of algorithms, hardware, and software stacks.

As Nvidia makes significant strides in the Datacenter business, our team holds a central role in maximizing the efficiency of our exponentially growing datacenter deployments and establishing a data-driven approach to hardware design and system software development. We collaborate extensively with diverse teams at Nvidia, spanning deep learning research and framework development teams, to silicon architecture. Thriving in such a high-impact, interdisciplinary environment necessitates not only technical proficiency but also a growth mindset and a pragmatic attitude — qualities that fuel our collective success in shaping the future of datacenter technology.

What You'll Be Doing:

  • Keeping abreast of the latest advancements in generative AI research.
  • Implementing and benchmarking cutting-edge large language and multi-modal models, analyzing their emergent patterns and system performance bottlenecks.
  • Pioneering the development of innovative optimizations to effectively lower the operational costs associated with deploying generative AI models at large scale.
  • Collaborating closely with production teams to incorporate the latest research advancements into cutting-edge hardware and/or software frameworks.

What We Need to See:

  • A minimum qualification of a Master's degree (or equivalent experience) in Computer Science, Artificial Intelligence, Applied Mathematics, or related fields. PhD a plus
  • A strong foundation in deep learning, with a particular emphasis on generative models.
  • 6+ years of relevant software development experience in modern deep learning frameworks such as PyTorch.
  • Growth mindset and pragmatic attitude.

Ways to Stand Out From the Crowd:

  • Published research or noteworthy contributions to the field of deep learning, particularly around algorithm development for efficient deployment at scale.
  • Experience with benchmarking accuracy and performance of AI models.
  • Experience with CUDA programming and GPU performance optimizations.
  • Familiarity with computer architecture and how it relates to AI algorithms development.

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 176,000 USD - 333,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.

Company Info.

NVIDIA

NVIDIA’s invention of the GPU sparked the PC gaming market. The company’s pioneering work in accelerated computing—a supercharged form of computing at the intersection of computer graphics, high performance computing and AI—is reshaping trillion-dollar industries, such as transportation, healthcare and manufacturing, and fueling the growth of many others.

  • Industry
    Cloud computing,Video games,Computer software,Semiconductors,Computer hardware,Consumer electronics,Artificial intelligence
  • No. of Employees
    22,473
  • Location
    2701 San Tomas Expressway, Santa Clara, CA 95050, USA
  • Website
  • Jobs Posted

Get Similar Jobs In Your Inbox

NVIDIA is currently hiring Senior AI Machine Learning Engineer Jobs in Santa Clara, CA, USA with average base salary of $176,000 - $333,500 / Year.

Similar Jobs View More