Senior Deep Learning Software Engineer

We are now looking for a Senior Deep Learning Software Engineer, DLSim. NVIDIA is seeking a Deep Learning Software Engineer to contribute to the creation of a compiler-oriented simulation infrastructure that swiftly assesses the forthcoming AI-accelerating GPU hardware and software advancements.

The DLSim Teams' core mission is to deliver full-stack simulation infrastructure for deep learning applications across a spectrum of GPUs. We actively collaborate with architecture, software, product, and research teams to shape and refine the strategic roadmap of DL hardware and software.

What you'll be doing:

Pioneer a novel deep learning compiler and simulation infrastructure that enables fast simulation of NVIDIA GPUs within DL compilers

Enhance DL compiler GPU kernel code generation with simulation-driven heuristics or cost models for optimal efficiency

Build a high-fidelity DL simulator with remarkable simulation speed

Partner across functional teams to understand and evaluate utilization, analyze, and identify opportunities for new or enhanced designs

Engage with Open Source communities

What we need to see:

A Masters or PhD degree in Computer Science, Computer Engineering, or a related STEM field (or equivalent experience)

3+ years of relevant work experience

Programming fluency in C/C and Python

In-depth understanding of compiler design and construction, particularly optimizations on code generation

Experience with architectural simulator design

Advanced knowledge of GPU and/or other AI accelerators

Effective communicator with solid prioritization skills, and a logical approach to problem-solving

Ways to stand out from the crowd:

A PhD in Computer Science and Engineering with a specialization in Computer Architecture, Compilers, or simulation techniques

Experience with the MLIR or LLVM compiler infrastructure and DL compilers such as OpenAI Triton and IREE

Made contributions to open-source projects such as MLIR or LLVM

Worked with the challenges of DL acceleration before, and ready to put your innovative ideas into motion and drive for success

Intelligent machines powered by AI computers that can learn, reason and interact with people are no longer science fiction. GPU-accelerated Deep Learning has provided the foundation for machines to learn, perceive, reason and solve problems. NVIDIA's GPUs run AI algorithms, simulating human intelligence, and act as the brains of computers, robots and self-driving cars that can perceive and understand the world. Increasingly known as "the AI computing company", and widely considered to be one of the technology world's most desirable employers, NVIDIA offers highly competitive salaries and a comprehensive benefits package.

Are you creative, motivated, and love a challenge? If so, we want to hear from you. Come, join our DLSim team, where you can help build real-time, cost-effective computing platforms driving our success in this exciting and rapidly growing field.

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