AMD is looking to accelerate AI workloads by enabling, optimizing, and integrating AI Software models across their GPU platform, ensuring efficient and scalable performance for training and real-time tasks.
Requirements
- Expertise in optimizing real-time AI models for deep learning applications (computer vision, NLP, etc.).
- Proficiency with AI frameworks (TensorFlow, PyTorch, ONNX Runtime, JAX, Triton) and their optimization for GPU architectures.
- Strong background in optimizing software for AMD GPUs or similar high-performance platforms.
- Familiarity with ROCm is a plus.
- Proven experience with performance optimization, benchmarking, and scaling AI models on GPUs.
Responsibilities
- Direct efforts to optimize AI frameworks for seamless compatibility and performance on AMD GPUs (Instinct, Navi).
- Oversee benchmarking, performance tuning, and optimization of AI Software odels to improve latency, throughput, and efficiency on AMD hardware.
- Partner with hardware, software, and QA teams to ensure tight integration of AI frameworks with ROCm for maximum performance.
- Drive AI model optimization innovations, enhancing the speed, efficiency, and scalability of AI workloads.
- Lead the vision and strategy for optimizing AI Software on AMD GPUs.
- Collaborate with customers and open-source communities to ensure that AMD’s AI solutions meet industry needs, fostering contributions to MIGraphX, vLLM, and other AMD AI Framework Softwareteams.
- Oversee automation frameworks to streamline model integration and performance testing, ensuring scalability across diverse AI workloads.
Other
- Leadership & Team Management: Lead and develop teams responsible for AI software model enablement and optimization, driving growth and success within the department.
- The Person: To be successful in this role, you will be a hands-on leader with strong technical skills and demonstrated success shipping software in an Agile environment.
- You are passionate about hiring and developing your team as well as planning and managing key engineering initiatives.
- You have outstanding people leadership skills to lead, motivate, and guide world class engineers working in a fast-paced, matrixed, global environment.
- Collaboration & Strategy: Exceptional ability to collaborate cross-functionally and define long-term strategies for AI/ML innovation.
- Communication: Strong verbal and written communication skills, with experience presenting to senior leadership and working with customers and partners.
