About the position
At AMD, we care deeply about transforming lives with our technology to enrich our industry, our communities, and the world. Our mission is to build great products that accelerate next-generation computing experiences - the building blocks for the data center, artificial intelligence, PCs, gaming, and embedded systems. We push the limits of innovation to solve the world's most important challenges, striving for execution excellence while being direct, humble, collaborative, and inclusive of diverse perspectives. In the ever-evolving landscape of artificial intelligence, we are a powerhouse - a cutting-edge 'AI Software Development Team'. Specialized in AI optimization, we fine-tune large language models to unlock unprecedented Generative AI efficiency. Our expertise extends beyond the hardware realm, encompassing 3P enablement, where we develop custom AI Software Solutions for industry-leading AI customers.
Responsibilities
- Work closely with strategic customers to understand their requirements and challenges and identify opportunities for AMD hardware and software to provide value.
- Close gaps in the AMD software stack needed to support customer solutions.
- Work hands-on as a technical expert and creative problem-solver, developing side-by-side with customers to drive projects from proof of concept to production.
- Act as the voice of the customer, translating their needs and insights into actionable feedback that shapes the future of AMD's AI roadmap.
- Thrive in unfamiliar territory with a high degree of autonomy, finding novel ways to apply AMD technology to real-world problems.
Requirements
- Strong programming skills in C/C++ and Python.
- Experience with GPU kernel programming using CUDA, HIP, or OpenCL.
- A track record of direct client engagement with experience working with customers to solve ambiguous technical problems.
- Strong performance analysis and optimization skills for both CPU and GPU.
- Experience with containerization and orchestration technologies like Singularity, Docker, and/or Kubernetes.
- Expertise with modern AI/ML frameworks (e.g., PyTorch, TensorFlow, JAX).
- Experience with distributed training and inference frameworks.
- Experience with open-source software development including collaboration with community maintainers and submitting contributions.
- Experience with software engineering methodologies such as Agile, Scrum, Kanban.
- Excellent analytical and problem-solving skills.
- Ability to work independently and as part of a team.
- BS with 5+ years of related experience or MS/PhD with 3+ years of related experience in Computer Science or related equivalent.
Nice-to-haves
- Experience in compiler, ISA.
- Experience shipping software in an end-customer production environment.
- Experience in implementing and optimizing communication primitives on GPU accelerators (NCCL/RCCL, OpenMP, MPI).
- Experience in all phases of software development, from requirement gathering, analysis, design, development, testing to final release.
- Experience providing clear and timely communication related to status and other key aspects of the project to the executive leadership team.
Benefits
- AMD benefits at a glance.
