Senior Software Engineer - Compilers and Applied AI

NVIDIA•Austin, TX

1d•$184,000 - $287,500

About The Position

NVIDIA has been transforming computer graphics, PC gaming, and accelerated computing for more than 25 years, a legacy fueled by great technology and amazing people. Today, the company is tapping into the unlimited potential of AI to define the next era of computing, where GPUs act 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. The team builds solutions to optimize all layers of the CUDA ecosystem, leading to class-leading speedups in modern high-performance workloads and models. NVIDIA is looking for an outstanding Senior Software Engineer that can architect and implement highly scalable solutions to different use-cases. As a member of the team, you will develop new innovative workflows, work on compiler- or runtime-driven solutions that accelerate critical workloads, generate optimal code patterns at scale, and other high-impact AI challenges. You will collaborate closely with internal NVIDIA software and hardware teams to push the latest developments into NVIDIA products.

Requirements

Bachelor’s degree in Computer Science, Electrical Engineering, or related field (or equivalent experience)
6+ years of industry or academia experience with software engineering, compilers and developer tools; exposure to building comprehensive optimization frameworks, and hands-on experience with product environments.
Strong knowledge of compilers, code generation, and GPU architecture.
Experience with GPU programming and performance optimization (CUDA or equivalent).
Extensive Python programming skills, along with software engineering fundamentals.
Basic programming skills in other languages such as C/C++, Racket and Rust.
Strong mathematical and scientific foundation relevant to optimization heuristics/algorithms, ML and data science.
Track record developing and productizing software, optimization frameworks and/or developer tooling.

Nice To Haves

Familiarity with genetic/evolutionary algorithms, predictive modeling, and complex systems.
Deep expertise in GPU performance optimizations, evidenced by benchmark wins or published results.
Hands-on experience building compilers or compiler components using the LLVM framework, including optimization passes and code generation.
Familiarity with NVIDIA and open source compilers like LLVM, MLIR, PTX and OpenAI Triton.
Experience with Data Science projects, specifically with MLOPS workflows and tools, like W&B, MLflow, etc.

Responsibilities

Design and build high-performance optimization frameworks for the entire CUDA ecosystem.
Co-design novel solutions with software, hardware and algorithm teams; influence and adopt new capabilities as they become available.
Develop reproducible, high-fidelity evaluation frameworks covering performance, quality and developer productivity.
Collaborate across the AI stack — from hardware through compilers/toolchains, kernels/libraries, frameworks, distributed training, and inference/serving.