Senior AI Compiler Engineer, MLIR

What you'll do

• Develop MLIR-based graph representations and optimizations for future GPU architectures.
• Define APIs and MLIR dialects for the AI compiler.
• Implement compiler optimizations and kernel generation for neural networks.
• Conduct performance optimizations and analysis for the inference engine.
• Partner with framework and hardware teams to enable new model patterns.
• Support Ahead-of-Time and Just-in-Time usability across data center and edge environments.
• Contribute to general software engineering tasks including debugging and testing.

What we're looking for

• Bachelor's, Master's, or Ph.D. in Computer Science, Computer Engineering, a related field, or equivalent experience.
• 3+ years of relevant work or research experience in performance analysis and compiler optimizations.
• Experience with compiler technologies such as MLIR, XLA, and LLVM.
• Excellent C/C++ and Python programming and software design skills, including debugging, performance analysis, and testing.
• Ability to work independently, define project goals and scope, and lead your own development efforts.
• Strong interpersonal skills and the ability to thrive in a fast-moving, dynamic, product-oriented team.
• Understanding of deep learning models, algorithms, and frameworks such as PyTorch and JAX (preferred).
• Experience with GPU kernel generation, CUDA, or OpenCL programming (preferred).