1 day ago
Austin, TX, USA +3 moreMid Level
H1B sponsor
Base Salary
$116k - $190k/yr
Responsibilities
- Develop and maintain compiler verification software and infrastructure for graph transformations, IR lowering, and GPU execution.
- Analyze and validate deep learning compiler optimizations, including graph rewrites, fusion passes, and mixed-precision transformations.
- Implement and extend test-generation systems covering model topologies, precision modes, and hardware targets.
- Improve measurement of functional quality and performance as models, compiler stacks, and hardware evolve.
- Investigate correctness issues, compiler-generated code, model behavior, and edge-case failures in collaboration with teammates.
Requirements
- Bachelor’s, master’s, or doctoral degree in Computer Science, Computer Engineering, Mathematics, or equivalent experience.
- At least 2 years of hands-on software engineering experience in compiler development, deep learning systems, or compiler verification.
- Proficiency in Python or C++ and experience with a major deep learning framework such as PyTorch, JAX/XLA, or TensorRT.
- Experience involving model execution, graph representation, or runtime behavior.
- Strong problem-solving and debugging skills with the ability to reason about model behavior and compiler-generated code.
- Preferred experience with LLVM, MLIR, TVM, XLA, compiler passes, IR semantics, correctness testing, formal methods, language specification, type systems, program semantics, proof-based verification, quantization, operator fusion, mixed precision, or graph-level optimization.
Benefits
- Eligible for equity and benefits.
- Applications will be accepted at least until November 4, 2026.
About Nvidia
Nvidia designs and sells GPUs and accelerated computing platforms for data centers, AI/ML, graphics, gaming, and automotive, monetizing through hardware, software platforms (CUDA, AI frameworks), and systems like DGX and networking. Customers include cloud providers, enterprises, researchers, and OEMs. Founded in 1993 and headquartered in Santa Clara, it is a public company traded on NASDAQ under NVDA.