Advanced Microdevices Pvt. Ltd. (India) · 12 hours ago
Principal Software Engineer – AI Compiler Engineer (NPU)
San Jose, CA
Full-time
Onsite
Lead/Staff
12+ years expAdvanced Micro Devices, Inc is a company focused on building innovative products for next-generation computing experiences. They are seeking a Principal Software Engineer to lead technical initiatives in the development of their NPU compiler stack, requiring deep expertise in compiler design and performance engineering.
BiotechnologyIndustrialPharmaceuticalManufacturingBiopharma
No H1B
Responsibilities
Provide technical leadership for the architecture and development of AMD’s NPU compiler stack
Design advanced IR transformations, graph‑level optimizations, and hardware‑aware fusion passes
Guide system-wide performance strategies using deep knowledge of HW architecture constraints (memory, parallelism, dataflows)
Collaborate with hardware architects, runtime teams, and ML framework teams (ONNX, PyTorch) to drive co‑design
Drive performance investigations, benchmarking, and software‑hardware co‑optimization
Represent AMD externally through technical publications, open‑source contributions, or industry engagement
Mentor MTS/SMTS engineers and set technical direction for large compiler initiatives
Qualification
Required
12+ years of experience in compilers, performance engineering, or system-level software
Strong C++ expertise and proficiency in Python
Extensive hands-on experience with LLVM and MLIR
Experience designing optimizations tightly coupled to hardware architectures (e.g., tensor accelerators, GPUs, NPUs)
Deep understanding of compiler internals: IR design, scheduling, memory optimizations, multi-stage lowering
Demonstrated experience delivering production-quality compiler or systems software
Experience driving technical strategy across teams and influencing product roadmaps
Preferred
Contributions to open‑source compiler/ML communities (LLVM, MLIR, XLA, TVM, Triton)
Experience with AI/ML model execution, graph optimization, and performance tuning
Experience with domain-specific languages or IRs for machine learning
Knowledge of NPU/GPU execution models, memory hierarchies, and parallelism strategies
Benefits
AMD benefits at a glance.
Company
Advanced Microdevices Pvt. Ltd. (India)
Advanced Microdevices (mdi) is a leader in innovative membrane technologies.
501-1000 employees
Funding
Current Stage
Late StageLeadership Team
Nalini Kant Gupta
Founder & Managing Director
Recent News
2024-10-18
2024-10-16
Company data provided by crunchbase