Advanced Microdevices Pvt. Ltd. (India) ยท 1 day ago
Sr. Staff Software Engineer - NPU Compiler Solutions
San Jose, CA
Full-time
Hybrid
Senior Level, Lead/StaffAdvanced Micro Devices, Inc is dedicated to building innovative products that enhance computing experiences across various platforms. They are seeking a Senior Staff Software Engineer to contribute to the architecture and design of the MLIR-centric NPU compiler platform, develop solutions for algorithm deployment on spatial compute devices, and collaborate with cross-functional teams on problem-solving and project planning.
BiopharmaBiotechnologyIndustrialManufacturing
No H1B
Responsibilities
Contributing to the architecture and design of the MLIR-centric NPU compiler platform
Develop and integrate solutions to efficiently deploy algorithms on spatial compute devices, such as NPUs
Work with cross functional teams to identify problems and create solutions
Work with management team on project planning activities
Qualification
Required
Strong C/C++ systems programming (modern C++), Python for tooling/prototyping
Hands-on MLIR experience: Dialect design (ops/types/attributes/interface), Pass pipelines, (pattern rewrites, canonicalization/legalization). Proficiency with core dialects: Linalg/Tensor, Affine, SCF, Vector
Accelerator compiler experience: GPU/NPU/AI engines or similar spatial devices; memory hierarchies, streams, scratchpads, NoC-aware optimization
Auto-optimization: practical experience building or integrating auto-tiling, auto-scheduling, and auto-tuning systems
Operator fusion and graph-level optimization for DSP&ML workloads (CNNs, transformers, FFT), layout/dtype transforms, quantization-aware workflows
Solid understanding of heterogeneous runtime models, concurrency, synchronization, and performance profiling
Master's or PhD in Electrical Engineering, Computer Science, or related field
Preferred
Polyhedral/affine analysis, dependence analysis, bank/conflict modeling, memory placement strategies
Open-source contributions to LLVM/MLIR or peer-reviewed publications in compilers/ML systems
ML compiler stacks: IREE, TVM, XLA, Glow, Halide; auto-schedulers/tuners integration
Expertise in Linux kernel/driver development for multi-processor heterogeneous systems
Knowledge of Acceleration platforms like GPU, TPU, APU, FPGAs
Benefits
AMD benefits at a glance.
Company
Advanced Microdevices Pvt. Ltd. (India)
Advanced Microdevices (mdi) is a leader in innovative membrane technologies.
Founded in 1976
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