Sr AI Software Development Engineer

AMD is a company focused on building innovative products that enhance computing experiences across various domains. They are seeking a highly motivated AI Software Development Engineer to join their team, enabling software for next-generation GPU computational accelerators and optimizing performance for AI applications. Responsibilities: Enable DL models, libraries, and applications for Instinct GPUs in both on-prem and Cloud environments; Work with a team of Software Engineers to optimize the software ecosystem for next-generation GPU computational accelerators; Analyze and optimize the performance of AI software and understand hardware bottlenecks to harness performance to hit close to roofline Qualifications: Strong programming skills in C++ and Python; Strong development experience in at least one major DL framework such as Pytorch or Tensorflow in inference, fine tuning and/or training; MS or PhD in Computer Science or Computer Engineering or related equivalent; Experience developing software and system-level performance optimizations with a solid architecture understanding in GPUs a plus; Experience with open-source software development including collaboration with community maintainers and submitting contributions is a plus; Publications in reputed peer-reviewed ML conferences/journals a plus; Excellent analytical and problem-solving skills root-causing/addressing performance issues; Ability to work independently and as part of a team; Willingness to learn skills, tools, and methods to advance the quality, consistency, and timeliness of AMD software products Required Skills: C++, Python, Deep Learning frameworks, PyTorch, TensorFlow, GPU architecture, AI software performance optimization, Open-source software development, Profiling tools, Parallel programming on GPUs, NCCL, RCCL, OpenMP, MPI, Containerization tools, Singularity, Docker, Kubernetes