Research Engineer - Accelerator Architecture

Job Description:

NVIDIA has been transforming computer graphics, PC gaming, and accelerated computing for more than 25 years. It's a unique legacy of innovation that's fueled by great technology-and amazing people. Today, we're tapping into the unlimited potential of AI to define the next era of computing. An era in which our GPU acts as the brains of computers, robots, and self-driving cars that can understand the world. Doing what's never been done before takes vision, innovation, and the world's best talent. As an NVIDIAN, you'll be immersed in a diverse, supportive environment where everyone is inspired to do their best work. Come join the team and see how you can make a lasting impact on the world.

Here in the NVIDIA hardware accelerator team, we work on new architectures to support novel application areas and AI models. Our work includes researching the evolving landscape of AI applications and models, analyzing underlying model architectures, and building implementations for various NVIDIA hardware accelerators such as LPUs. We analyze mappings to existing and future accelerator systems, model performance, and work multi-functionally with the hardware design team on novel hardware features e.g. functional units, numeric modes, interconnect, system integration, etc. to unlock new application areas for NVIDIA accelerators. There are opportunities to participate in a wider range of R&D activities, either internally or externally with key NVIDIA partners.

What you'll be doing:
* AI application and model research
* Performance modeling
* Multi-functional work with hardware and software teams
* Next generation hardware architecture development
* Support internal and outward-facing R&D

What we need to see:
* MSc or higher degree in CS/EE/CE/Mathematics or equivalent experience
* Minimum 2 years of relevant experience, preferably in AI models and applications
* Good foundation in computer science
* Knowledge of LLMs and other Gen AI applications
* Solid understanding of computer architecture and computer arithmetic
* Python and common ML frameworks such as PyTorch & TensorFlow
* Experience with performance analysis / modelling
* Problem solving mentality

Ways to stand out from the crowd:
* Experience with scientific computing & HPC
* Prior experience in optimizing applications on specialized accelerators (GPU, FPGA, or other custom accelerators).
* Experience with compiler tools and MLIR.
* Experience in delivering complex projects in a fast paced environment.

We are an equal opportunity employer and value diversity at our company. We do not discriminate on the basis of race, religion, color, national origin, sex, gender, gender expression, sexual orientation, age, marital status, veteran status, or disability status. We will ensure that individuals with disabilities are provided reasonable accommodation to participate in the job application or interview process, to perform essential job functions, and to receive other benefits and privileges of employment. Please contact us to request accommodation.

Widely considered to be one of the technology world's most desirable employers, NVIDIA offers highly competitive salaries and a comprehensive benefits package. As you plan your future, see what we can offer to you and your family www.nvidiabenefits.com