AI Compiler Engineer - Remote (US & Canada)

BEARCloud's client is seeking an AI Compiler Engineer developing next-generation AI inference accelerator technology. This role will focus on building and optimizing graph compiler technology for AI/ML workloads, improving model performance, latency, and hardware utilization. What You'll Do • Develop and optimize AI/ML graph compilers for inference accelerators. • Implement graph-level optimizations including layer/operator fusion, quantization, tiling, and graph transformations. • Build compiler passes that convert PyTorch/TensorFlow models into intermediate representations (IR). • Work with MLIR, Torch-FX, parsing, semantic analysis, IR generation, and code generation. • Partner with AI researchers and hardware architects to optimize models for GPU, TPU, ASIC, or custom accelerator architectures. • Analyze performance and identify opportunities to improve latency, throughput, and hardware utilization. • Provide technical guidance and mentorship to other engineers. What We're Looking For • 3+ years of compiler development experience, ideally with AI/ML graph compilers. • Hands-on experience with MLIR, Torch-FX, LLVM, or similar compiler frameworks. • Strong understanding of intermediate representations (IR), compiler passes, graph transformations, and code generation. • Experience with fusion, quantization, tiling, and neural network operators. • Knowledge of GPU, TPU, ASIC, NPU, or AI accelerator architectures. • Strong C++ and/or Python development skills. • BS/MS in Computer Science, Electrical Engineering, or related field; PhD a plus. • Open-source AI/compiler contributions and technical leadership experience are a plus.