Advanced Microdevices Pvt. Ltd. (India) · 10 hours ago
Performance Engineer (ML)
San Jose, CA
Full-time
Onsite
Senior LevelAdvanced Micro Devices, Inc (AMD) is dedicated to building innovative products that enhance computing experiences across various domains including AI and data centers. The company is seeking an experienced ML Compiler Engineer to optimize performance for machine learning workloads on their datacenter GPU infrastructure, focusing on compiler technology and efficient code generation.
BiopharmaBiotechnologyIndustrialManufacturing
No H1B
Responsibilities
Identify and analyze performance bottlenecks in ML workloads at the compiler level, using profiling tools and performance analysis methodologies
Design and implement optimization passes in the LLVM backend to improve code generation quality for datacenter GPU architectures
Develop and enhance instruction selection, scheduling, and register allocation strategies specific to ML computation patterns
Collaborate with hardware architecture teams to ensure compiler optimizations align with and fully exploit GPU microarchitectural features
Contribute bug fixes, optimizations, and new features to upstream LLVM projects, following open source development practices
Maintain and improve internal LLVM-based compiler toolchains, ensuring compatibility with upstream changes while adding proprietary enhancements
Create comprehensive benchmarks and test suites to validate compiler optimizations and prevent performance regressions
Document compiler optimizations, architectural decisions, and best practices for both internal and external audiences
Work with ML framework teams to understand and optimize compilation of real-world models and workloads
Participate in code reviews, design discussions, and technical presentations to share knowledge across the organization
Qualification
Required
Master's or Ph.D. in Computer Science, Computer Engineering, or a related field, with multiple years of professional experience in compiler development, with a significant focus on LLVM or other modern compiler backend
Preferred
Strong background in computer architecture and software engineering practices
Experience designing and working with compiler backends
Active contribution history to the open-source LLVM project
Demonstrated mastery of LLVM backend components, including LLVM IR, Machine IR (MIR), SelectionDAG, GlobalISel, Register Allocation, TableGen, and Instruction Scheduling
Effective communication and problem-solving skills
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