Description
We're building the next generation of compiler technologies to accelerate deep learning workloads. You'll implement compiler verification software and related infrastructure in the AI space, working alongside diverse minds in GPU computing and systems software.
Responsibilities
- Develop and maintain components of systems to reason about correctness in deep learning compilers, across graph transformations, IR lowering, and GPU execution.
- Work with deep learning compiler and architecture teams to analyze and validate optimizations, ensuring they preserve semantics and numerical behavior.
- Implement and extend test generation systems that use deep learning solutions and analysis methods to drive in-depth testing.
- Contribute to improving how we measure functional quality and performance as models, compiler stacks, and hardware continue to evolve.
Requirements
- BS, MS or PhD in Computer Science, Computer Engineering, Mathematics, or equivalent experience.
- 2+ years of hands-on software engineering experience, with experience in compiler development, deep learning systems, or compiler verification.
- Proficiency in Python or C++ and experience with one major DL framework (PyTorch, JAX/XLA, TensorRT, etc.).
- Strong problem-solving and debugging skills , ability to reason about model behavior and compiler-generated code, investigate correctness issues, and collaborate with teammates to resolve edge-case failures.
Preferred Qualifications
- Experience with compiler technologies such as LLVM, MLIR, TVM, or XLA, and familiarity with compiler passes, IR semantics, or correctness testing.
- Exposure to formal methods or language specification: type systems, program semantics, or proof-based verification.
- Experience with DL model internals such as quantization, operator fusion, mixed-precision, or graph-level optimization.
Your base salary will be determined based on your location, experience, and other objective job-related factors. The expected base salary range for this position is 116,000 USD – 189,750 USD. You will also be eligible for equity and benefits.