Senior AI Compiler Engineer, MLIR

This is a fully remote position, open to applicants in California, +2 more states.

📋 Description

• Create MLIR-based graph representations and enhancements for future GPU architectures.

• Collaborate with framework and hardware teams to facilitate new model patterns and emerging GPU architectural features.

• Establish APIs and MLIR dialects.

• Perform performance optimizations and analyses.

• Execute compiler optimizations and kernel generation for neural networks.

• Contribute to overall software engineering tasks.

• Set project objectives and define scope.

• Drive personal development initiatives.

• Mentor junior engineers and interns.

⛳️ Requirements

• Bachelor's, Master's, or Ph.D. in Computer Science, Computer Engineering, a related field, or equivalent experience.

• Over 3 years of relevant experience in performance analysis and compiler optimizations.

• Familiarity with compiler technologies such as MLIR, XLA, and LLVM.

• Strong programming and software design skills in C/C++ and Python.

• Experience in debugging, performance analysis, and testing.

• Capacity to work independently, set project goals and scope, and lead personal development efforts.

• Excellent interpersonal skills and the ability to excel in a fast-paced, dynamic, product-oriented environment.

• Knowledge of deep learning models, algorithms, and frameworks like PyTorch and JAX.

• Experience with GPU kernel generation aimed at achieving high performance and rapid build times.

• Proficiency in GPU architecture, including experience with CUDA or OpenCL programming.

🏝️ Benefits

• Competitive salaries.

• Generous benefits package.

• Equity.

• Additional benefits.