Senior Research Engineer - Enterprise Products

Remote: 
Full Remote
Contract: 

Offer summary

Qualifications:

Bachelor's degree or equivalent experience in Computer Science or related field., 8+ years of industry experience in Deep Learning frameworks such as PyTorch or TensorFlow., Strong programming skills in C++ and Python, with contributions to major open-source projects., Understanding of Machine Learning techniques, algorithms, and data structures..

Key responsibilities:

  • Develop new models and algorithms for Large Language Models and Natural Language Processing.
  • Design and implement multi-node serving architectures for distributed LLM inference.
  • Optimize inference serving systems using advanced quantization techniques and latency optimization strategies.
  • Collaborate with engineering teams to ensure seamless integration of software across NVIDIA's serving stack.

NVIDIA logo
NVIDIA XLarge http://www.nvidia.com
10001 Employees
See all jobs

Job description

We are now looking for a Senior Research Engineer passionate about Generative AI inference. Are you excited to change the way people infuse AI into products and services? NVIDIA is at the forefront of generative AI models, from language to images. NVIDIA provides building blocks to democratize AI and make generative AI easy to develop, integrate, and deploy. Our team is dedicated to developing optimized inferencing technologies to support our growing generative AI needs. We contribute to all steps of the machine learning lifecycle: from conceptualization, to applied research, engineering for optimized inference, and deployment. Collaborate with research teams, engineers, and open-source community. Implement optimized LLM algorithms.

What you will be doing:

  • Developing new models and algorithms focused on Large Language Models, Natural Language Processing, and Deep Learning.

  • Design and implement multi-node serving architectures disaggregated serving and distributed LLM inference

  • Optimize multi-LoRA (and other PEFT technique) inference serving systems

  • Apply sophisticated quantization techniques (FP4/INT4, FP8) to reduce model footprint while preserving quality

  • Implement speculative decoding (draft target, eagle, medusa etc) and other latency optimization strategies

  • Demonstrating good engineering practices and mentoring other team members to do the same.

  • Collaborating with engineering teams across all of NVIDIA to ensure our software integrates seamlessly up and down the NVIDIA accelerated serving stack.

What we need to see:

  • Understanding of modern techniques in Machine Learning, Deep Neural Networks, Natural Language Processing, or Speech Recognition.

  • 8+ years of industry experience in Deep Learning frameworks (PyTorch or TensorFlow).

  • Passion for software engineering, especially with excellent C++ and Python development skills, with meaningful contributions to major open-source projects.

  • Strong communication and interpersonal skills, along with the ability to work in a dynamic and distributed team. A history of mentoring junior engineers and interns is a huge plus.

  • Bachelor's degree or equivalent experience.

  • A desire to constantly grow and learn new things.

  • Strong computer science fundamentals - algorithms and data structures, computational complexity, parallel and distributed computing, system software.

Ways to stand out from a crowd:

  • Experience architecting or developing large-scale distributed systems for deep learning.

  • Knowledge of CPU and/or GPU architecture.

  • GPU programming (CUDA).

The base salary range is 184,000 USD - 356,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.

Required profile

Experience

Spoken language(s):
English
Check out the description to know which languages are mandatory.

Other Skills

  • Mentorship
  • Social Skills
  • Teamwork
  • Communication
  • Problem Solving

Related jobs