Senior Software Engineer - Compilers and Applied AI

at Nvidia
📍 Austin, United States
USD 184,000-356,500 per year
SENIOR
✅ On-site

SCRAPED

Used Tools & Technologies

Not specified

Required Skills & Competences ?

Python @ 4 Algorithms @ 7 MLOps @ 4 Data Science @ 7 Performance Optimization @ 4 MLFlow @ 4 Rust @ 4 CUDA @ 4 GPU @ 7

Details

NVIDIA has been transforming computer graphics, PC gaming, and accelerated computing for more than 25 years. Today we are tapping into the unlimited potential of AI to define the next era of computing. We build solutions to optimize all layers of the CUDA ecosystem, leading to class-leading speedups in modern high-performance workloads and models. As a member of the team you will develop innovative workflows, work on compiler- or runtime-driven solutions that accelerate critical workloads, generate optimal code patterns at scale, and collaborate closely with internal NVIDIA software and hardware teams to push the latest developments into NVIDIA products.

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.

Requirements

  • Bachelor’s degree in Computer Science, Electrical Engineering, or related field (or equivalent experience); MS or PhD preferred.
  • 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.

Ways to Stand Out

  • 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.

Compensation & Benefits

  • Base salary ranges (location- and level-dependent):
    • Level 4: 184,000 USD - 287,500 USD
    • Level 5: 224,000 USD - 356,500 USD
  • You will also be eligible for equity and benefits.

Other Details

  • Location: Austin, Texas, United States.
  • Employment type: Full time.
  • Applications for this job will be accepted at least until December 13, 2025.
  • NVIDIA is an equal opportunity employer and committed to fostering a diverse work environment.