[Remote] AI Compiler Engineer

Note: The job is a remote job and is open to candidates in USA. EnCharge AI is a leader in advanced AI hardware and software systems for edge-to-cloud computing. They are seeking a highly skilled AI Compiler Engineer to develop and optimize graph compilers for AI and ML workloads, collaborating with hardware architects and AI researchers to enhance performance and enable efficient model deployment.ResponsibilitiesArchitect, design, and implement optimizations for AI model execution on graph compilers to improve performance, reduce latency, and maximize hardware utilizationWork closely with ML researchers, hardware engineers, and software developers to design and deploy AI models, understanding and addressing hardware-specific challengesWork on performance optimizations for neural network models, such as layer fusion, operator fusion, and graph-level transformationsDevelop compiler optimizations and passes that convert high-level AI models (e.g., from TensorFlow, PyTorch) into intermediate representations (IR)Implement parsing, semantic analysis, and IR generation for deep learning frameworksResearch and integrate the latest advancements in compiler design, ML model optimizations, and hardware acceleration into graph compilersProvide leadership, mentorship, and technical guidance to a team of engineers focused on graph compiler optimizationsSkillsBachelor's or Master's degree in Computer Science, Electrical Engineering, or related field (Ph.D. preferred)3+ years in compiler development, with a strong focus on AI or ML graph compilersProficiency in AI graph compiler frameworks (e.g., MLIR, Torch-FX)Solid background in hardware architectures (e.g., GPUs, TPUs, ASICs) and optimization techniques such as fusion, quantization, and tilingFamiliarity with neural networks operators and code generationStrong understanding of intermediate representations, code parsing, and semantic analysis in compiler designProficiency in C++, Python, or other programming languages commonly used in compiler developmentDemonstrated experience leading and mentoring engineering teams with successful project deliveryPh.D. in a related fieldOpen-source contributions to AI software frameworks and libraries is a plusCompany OverviewEnCharge AI designs analog in-memory-computing AI chips and develops AI systems for AI computing. It was founded in 2022, and is headquartered in Santa Clara, California, USA, with a workforce of 11-50 employees. Its website is https://enchargeai.com.Company H1B SponsorshipEnCharge AI has a track record of offering H1B sponsorships, with 1 in 2026, 4 in 2025, 2 in 2024, 1 in 2023. Please note that this does not guarantee sponsorship for this specific role.