Senior Machine Learning Engineer, LLM

NVIDIA
Apply Now

Job Description

We are now looking for a Senior Machine Learning Engineer (LLM)! NVIDIA is seeking machine learning engineers that make LLMs faster and slimmer while minimizing training friction. You can (1) distill LLM research literature into its core, (2) run experiments at scale, (3) generate insights to support or refute the efficacy of a technique, and (4) generate reproducible training recipes.

Our mission is to create new value for NVIDIA's largest customers. We support this mission by building robust, reproducible, and friction-free training recipes that make compression-aware training and inference seamless. We own the top of the stack, and our largest customers directly interact with the artifacts that you will build.

What you'll be doing:

  • Keep abreast on research in LLM compression algorithms
  • Design, implement, and evaluate LLM compression algorithms that minimize regressions, reduce training friction, and improve performance
  • Run ablation experiments to understand the Pareto-frontier
  • Build robust, reproducible, and portable training recipes
  • Work with production SW teams to realize recipes in production workflows

What we need to see:

  • M.S. degree or equivalent experience in Computer Science or a related field, and/or proven track record. PhD preferred
  • 6+ years of relevant work experience
  • Strong programming skills and ability to debug ML systems
  • Experience with PyTorch
  • Proficient in the math of machine learning
  • Strong written and oral communication skills

Ways to stand out from the crowd:

  • Background with mixed-precision training flows
  • Experience working in Efficient LLMs
  • Experience with performance optimization of training and inference systems

GPU computing is the most productive and pervasive platform for deep learning and AI. It begins with the most advanced GPUs and the systems and software we build on top of them. We integrate and optimize every deep learning framework. We work with the major systems companies and every major cloud service provider to make GPUs available in data centers and in the cloud. We craft computers and software to bring AI to edge devices, such as self-driving cars and autonomous robots. AI has the potential to spur a wave of social progress unmatched since the industrial revolution.

NVIDIA is widely considered to be one of the technology world’s most desirable employers. We have some of the most experienced and hard-working people in the world working for us. Are you creative and collaborative? Do you love the challenge of defining and leading an industry If so, we want to hear from you

The base salary range is 180,000 USD - 339,250 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 Large Language Models Engineer Jobs in Santa Clara, CA, USA with average base salary of $180,000 - $339,250 / Year.

Similar Jobs View More