Director Software Engineering - AI/ML Compilers
AMD San Jose, CA Full-time 3/24/2026 $232k - $348k per year
PhD Entry-Level
Approval 98.6%•Total filings 728•New hires 184•
✓ Established Sponsor
•FY 2025Job Description
AMD is seeking a Director of Machine Learning to lead a team focused on developing efficient methods for Generative AI, particularly in distributed training on Neural Processing Units (NPUs). The role involves collaborating with various teams to optimize model performance and drive innovative solutions in AI/ML compiler technologies.
Requirements
- Experience with delivering ML compiler solutions
- Understanding of NPU hardware architecture
- Defining scalable algorithms for efficient problem mapping
- Interfacing with customers to understand requirements
- Driving a team to develop solutions
- Experience with ML frameworks such as Onnx, PyTorch, or TensorFlow
- Excellent C++ and Python programming skills
- Strong communication and problem-solving skills
- Strong leadership and management skills
- PhD or Master's degree in Computer Science, Artificial Intelligence, Machine Learning, or a related field
Responsibilities
- Lead a growing team focused on research, design, and implement novel methods for efficient Generative AI
- Develop ML compilers for low power inference architecture
- Define optimized tiling algorithms and fusion algorithms at different memory hierarchy
- Influence the direction of AMD AI platform
- Collaborate across teams with various groups and stakeholders
- Propose and apply innovative model quantization, sparsity, and acceleration algorithms
- Collaborate with software and hardware team to E2E co-optimize model performance
- Work with open-source frameworks and community to integrate AMD optimized models and libraries
Benefits
- AMD provides a competitive 'Total Rewards' package that focuses on financial growth, health, and work-life balance.
Is this job posting expired or no longer available?