We are now looking for a Senior Deep Learning Software Engineer, DLSim! NVIDIA is seeking aDeep Learning Software Engineerto 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’llbe 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 comprehensivebenefits 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 144,000 USD - 270,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.
