Senior AI Compiler Engineer, MLIR
Core
Building an MLIR-based AI compiler for NVIDIA's inference engine to optimize performance, build speed, and memory usage across data center and edge devices.
Role type
Senior IC AI Compiler Engineer
Builds
MLIR-based graph representations, optimizations, and kernel generation for neural networks
Domain
AI computing, GPU architecture, compiler technology
Deliverable
production ML models
Required skills
MLIR, XLA, LLVM, C/C++, Python, performance analysis, compiler optimizations, GPU kernel generation
Preferred skills
Deep learning frameworks (PyTorch, JAX), CUDA/OpenCL, GPU architecture knowledge, mentoring
Technologies
MLIR, XLA, LLVM, CUDA, OpenCL, PyTorch, JAX
Responsibilities
Develop MLIR-based graph representations and optimizations for future GPU architectures; Partner with framework and hardware teams to enable new model patterns; Define APIs and MLIR dialects; Conduct performance optimizations and analysis; Implement compiler optimizations and kernel generation for neural networks
Seniority
Senior, hands-on IC