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 compilersEnhance DL compiler GPU kernel code generation with simulation-driven heuristics or cost models for optimal efficiencyBuild a high-fidelity DL simulator with remarkable simulation speedPartner across functional teams to understand and evaluate utilization, analyze, and identify opportunities for new or enhanced designsEngage with Open Source communitiesWhat 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 experienceProgramming fluency in C/C++ and PythonIn-depth understanding of compiler design and construction, particularly optimizations on code generationExperience with architectural simulator designAdvanced knowledge of GPU and/or other AI acceleratorsEffective communicator with solid prioritization skills, and a logical approach to problem-solvingWays to stand out from the crowd:A PhD in Computer Science and Engineering with a specialization in Computer Architecture, Compilers, or simulation techniquesExperience with the MLIR or LLVM compiler infrastructure and DL compilers such as OpenAI Triton and IREEMade contributions to open-source projects such as MLIR or LLVMWorked with the challenges of DL acceleration before, and ready to put your innovative ideas into motion and drive for successIntelligent 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.
Job Title
Senior Deep Learning Software Engineer, DLSim