Senior Research Engineer - Enterprise Products

Nvidia Corporation in; Santa Clara (CA); On-site; USD 184,000 - 357,000; Full time

Job summary

NVIDIA seeks a Senior Research Engineer to innovate in Generative AI by developing new models and implementing optimized algorithms. This role requires strong technical acumen in ML frameworks, exceptional programming skills, and collaboration across teams to enhance AI products and services. Join a leader in AI to drive the development of next-gen technologies.

Benefits

Equity and benefits

Qualifications

• 8+ years industry experience in Deep Learning frameworks.
• Excellent C++ and Python development skills.
• Strong computer science fundamentals - algorithms, data structures.

Responsibilities

• Develop new models and algorithms for LLM and NLP.
• Design multi-node serving architectures for LLM inference.
• Optimize inference serving systems with sophisticated techniques.

Skills

Machine Learning

Deep Learning

Natural Language Processing

C++

Python

Communication

Interpersonal Skills

Mentoring

Education

Bachelor's degree or equivalent experience

Tools

PyTorch

TensorFlow

CUDA

Job description

Senior Research Engineer - Enterprise Products (Finance)

Senior Research Engineer - Enterprise Products (Finance)

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.